search

eriche2016 / image_caption_with_semantic_attenion

image caption with semantic attention
12 stars 8 forks source link

train.lua #10

Open liuxx2016 opened 6 years ago

liuxx2016 commented 6 years ago

When I run train.lua, an error occurs when iterates 39075 times. As follows： ... iter 39073: 3.309339
iter 39074: 3.151523
iter 39075: 3.371971
/home/XXX/torch/install/bin/luajit: ./misc_saver/BilinearD3_version2.lua:90: cublas runtime error : the GPU program failed to execute at /tmp/luarocks_cutorch-scm-1-5008/cutorch/lib/THC/THCBlas.cu:246 stack traceback: [C]: in function 'addmm' ./misc_saver/BilinearD3_version2.lua:90: in function 'accGradParameters' ...XXX/torch/install/share/lua/5.1/nngraph/gmodule.lua:479: in function 'neteval' ...XXX/torch/install/share/lua/5.1/nngraph/gmodule.lua:490: in function 'accGradParameters' /home/XXX/torch/install/share/lua/5.1/nn/Module.lua:32: in function 'backward' ./misc_saver/LanguageModel.lua:574: in function 'updateGradInput' /home/XXX/torch/install/share/lua/5.1/nn/Module.lua:31: in function 'backward' train.lua:399: in function 'lossFun' train.lua:452: in main chunk [C]: in function 'dofile' ...xiao/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk [C]: at 0x00405d50

What is wrong with this? How to solve it? THANKS