Hi,
After downloading MS COCO dataset and run prepo.py, when I try to train, the following error appears:
THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-5309/cutorch/lib/THC/generic/THCStorage.cu line=66 error=2 : out of memory
/home/linux/torch/install/bin/luajit: /home/linux/torch/install/share/lua/5.1/torch/File.lua:351: cuda runtime error (2) : out of memory at /tmp/luarocks_cutorch-scm-1-5309/cutorch/lib/THC/generic/THCStorage.cu:66
Any solutions? :(
P.S when I run it on CPU mode by passing the parameter -gpuid -1, it starts training. But I think it never ends!
Hi, After downloading MS COCO dataset and run prepo.py, when I try to train, the following error appears:
THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-5309/cutorch/lib/THC/generic/THCStorage.cu line=66 error=2 : out of memory /home/linux/torch/install/bin/luajit: /home/linux/torch/install/share/lua/5.1/torch/File.lua:351: cuda runtime error (2) : out of memory at /tmp/luarocks_cutorch-scm-1-5309/cutorch/lib/THC/generic/THCStorage.cu:66
Any solutions? :(
P.S when I run it on CPU mode by passing the parameter -gpuid -1, it starts training. But I think it never ends!