search

jcjohnson / densecap

Dense image captioning in Torch
MIT License
1.58k stars 432 forks source link

Out of memory when running the camera demo on a single machine #55

Open yosit opened 7 years ago

yosit commented 7 years ago

I'm on a mac 15" with nvidia, cuda, cudnn installed and getting this:

qlua webcam/single_machine_demo.lua loading checkpoint from data/models/densecap/densecap-pretrained-vgg16.t7 done loading checkpoint THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-8066/cutorch/lib/THC/generic/THCStorage.cu line=65 error=2 : out of memory qlua: /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:11: cuda runtime error (2) : out of memory at /tmp/luarocks_cutorch-scm-1-8066/cutorch/lib/THC/generic/THCStorage.cu:65 stack traceback: [C]: at 0x010c1f0870 [C]: in function 'resize' /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:11: in function 'torch_Storage_type' /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:57: in function 'recursiveType' /Users/yosit/torch/install/share/lua/5.1/nn/Module.lua:152: in function 'type' /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:45: in function 'recursiveType' /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:41: in function 'recursiveType' /Users/yosit/torch/install/share/lua/5.1/nn/Module.lua:152: in function 'type' /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:45: in function 'recursiveType' /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:41: in function 'recursiveType' /Users/yosit/torch/install/share/lua/5.1/nn/Module.lua:152: in function 'type' /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:45: in function 'recursiveType' /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:41: in function 'recursiveType' /Users/yosit/torch/install/share/lua/5.1/nn/Module.lua:152: in function 'type' /Users/yosit/torch/install/share/lua/5.1/nn/utils.lua:45: in function 'recursiveType' /Users/yosit/torch/install/share/lua/5.1/nn/Module.lua:152: in function 'type' ./densecap/DenseCapModel.lua:199: in function 'convert' webcam/single_machine_demo.lua:185: in function 'main' webcam/single_machine_demo.lua:256: in main chunk

any idea