@@ -9,7 +9,8 @@ def new_iterator(data, n_jobs, batch_size, repeat=True, shuffle=True, n_prefetch
n_processes=n_jobs,
n_prefetch=n_prefetch,
batch_size=batch_size,
- repeat=repeat, shuffle=shuffle)
+ repeat=repeat, shuffle=shuffle,
+ shared_mem=np.zeros((3,1024,1024), dtype=np.float32).nbytes)
else:
it = SerialIterator(data,