Explorar o código

setting default shared memory size. otherwise there may be some issues pre-computing it from the first batch

Dimitri Korsch %!s(int64=5) %!d(string=hai) anos
pai
achega
1987c10512
Modificáronse 1 ficheiros con 2 adicións e 1 borrados
  1. 2 1
      cvdatasets/utils/dataset.py

+ 2 - 1
cvdatasets/utils/dataset.py

@@ -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,
 			batch_size=batch_size,