RuntimeError: cuda runtime error (2) : out of memory at /opt/conda/conda about rfbnet HOT 4 CLOSED

@yinggo Your GPU memory is limited, use 300 input size or reduce your batch size.

from rfbnet.

@yinggo What's the version of your pytorch. We only support 0.4.0 now, and 0.4.1 maybe incompatible.

from rfbnet.

@yinggo What's the version of your pytorch. We only support 0.4.0 now, and 0.4.1 maybe incompatible.

Thanks! I changed to 0.4.0, now it works.

from rfbnet.

I meet a new error, can you give some suggestions? @ruinmessi
Initializing weights... Loading Dataset... Training RFB_E_vgg on VOC0712 THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1524584710464/work/aten/src/THC/generic/THCStorage.cu line=58 error=2 : out of memory Traceback (most recent call last): File "train_RFB.py", line 257, in <module> train() File "train_RFB.py", line 220, in train out = net(images) File "/home/ds1/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in __call__ result = self.forward(*input, **kwargs) File "/home/ds1/Downloads/RFBNet/models/RFB_Net_E_vgg.py", line 238, in forward x = self.base[k](x) File "/home/ds1/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in __call__ result = self.forward(*input, **kwargs) File "/home/ds1/anaconda3/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward self.padding, self.dilation, self.groups) RuntimeError: cuda runtime error (2) : out of memory at /opt/conda/conda-bld/pytorch_1524584710464/work/aten/src/THC/generic/THCStorage.cu:58 Exception ignored in: <bound method _DataLoaderIter.__del__ of <torch.utils.data.dataloader._DataLoaderIter object at 0x7f188a4780f0>> Traceback (most recent call last): File "/home/ds1/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 349, in __del__ self._shutdown_workers() File "/home/ds1/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 328, in _shutdown_workers self.worker_result_queue.get() File "/home/ds1/anaconda3/lib/python3.6/multiprocessing/queues.py", line 337, in get return _ForkingPickler.loads(res) File "/home/ds1/anaconda3/lib/python3.6/site-packages/torch/multiprocessing/reductions.py", line 70, in rebuild_storage_fd fd = df.detach() File "/home/ds1/anaconda3/lib/python3.6/multiprocessing/resource_sharer.py", line 57, in detach with _resource_sharer.get_connection(self._id) as conn: File "/home/ds1/anaconda3/lib/python3.6/multiprocessing/resource_sharer.py", line 87, in get_connection c = Client(address, authkey=process.current_process().authkey) File "/home/ds1/anaconda3/lib/python3.6/multiprocessing/connection.py", line 487, in Client c = SocketClient(address) File "/home/ds1/anaconda3/lib/python3.6/multiprocessing/connection.py", line 614, in SocketClient s.connect(address) ConnectionRefusedError: [Errno 111] Connection refused

Python 3.6
pytorch: 0.4.0
cuda 8.0
GPU: TITAN X.

from rfbnet.