Add capacity to the queues
@@ -156,7 +156,7 @@ if __name__ == "__main__":
optimizer = optim.Adam(net.parameters(), lr=LEARNING_RATE, eps=1e-3)
- train_queue = mp.Queue(maxsize=PROCESSES_COUNT)
+ train_queue = mp.Queue(maxsize=PROCESSES_COUNT*10)
data_proc_list = []
tgt_nets = []
for proc_idx in range(PROCESSES_COUNT):
