Error in fine tuning with 2 gpus

[ankitag9]

Hi,

I have a pretrained model and I am trying to fine tune it using 2 gpus. I am getting the following error -

/home/strange/torch/install/bin/luajit: /home/strange/torch/install/share/lua/5.1/nn/Container.lua:67: In 1 module of nn.Sequential: /home/strange/torch/install/share/lua/5.1/nn/Linear.lua:67: Assertion `THCTensor(checkGPU)(state, 4, r, t, vec1, vec2)' failed. at /home/strange/torch/extra/cutorch/lib/THC/generic/THCTensorMathBlas.cu:138 stack traceback: [C]: in function 'addr' /home/strange/torch/install/share/lua/5.1/nn/Linear.lua:67: in function </home/strange/torch/install/share/lua/5.1/nn/Linear.lua:53> [C]: in function 'xpcall' /home/strange/torch/install/share/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors' /home/strange/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward' train.lua:499: in function 'train_batch' train.lua:752: in function 'train' train.lua:1080: in function 'main' train.lua:1083: in main chunk [C]: in function 'dofile' ...ange/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk [C]: at 0x00405d50

WARNING: If you see a stack trace below, it doesn't point to the place where this error occurred. Please use only the one above. stack traceback: [C]: in function 'error' /home/strange/torch/install/share/lua/5.1/nn/Container.lua:67: in function 'rethrowErrors' /home/strange/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward' train.lua:499: in function 'train_batch' train.lua:752: in function 'train' train.lua:1080: in function 'main' train.lua:1083: in main chunk [C]: in function 'dofile' ...ange/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk [C]: at 0x00405d50

What can be done to remove this error?

[yoonkim]

try first converting the model to cpu using convert_to_cpu.lua

[ankitag9]

It works.. Thanks!!