Alice-k98

Results 2 comments of Alice-k98

我也有与2同样的问题,batch_size应该怎样修改呢

> 我用的pytorch是0.3.1,cuda8.0,ubuntu16 > Traceback (most recent call last): > File "/home/cv2018/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 333, in del > self._shutdown_workers() > File "/home/cv2018/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 319, in _shutdown_workers > self.data_queue.get() > File "/home/cv2018/anaconda3/lib/python3.6/multiprocessing/queues.py", line...