Description
This looks good. How do you run this on CPU? Is this hardcoded to run only on GPUs (NVIDIA)? When I tried removing reference to CUDA and try running on CPU it gives me an error saying:
/home/sanoob/torch/install/bin/luajit: ...ob/torch/install/share/lua/5.1/nn/SpatialConvolution.lua:100: bad argument #1 (field finput is not a torch.FloatTensor)
stack traceback:
[C]: in function 'SpatialConvolutionMM_updateOutput'
...ob/torch/install/share/lua/5.1/nn/SpatialConvolution.lua:100: in function 'updateOutput'
/home/sanoob/torch/install/share/lua/5.1/nn/Sequential.lua:39: in function 'updateOutput'
/home/sanoob/torch/install/share/lua/5.1/nn/Sequential.lua:39: in function 'updateOutput'
/home/sanoob/torch/install/share/lua/5.1/nn/Sequential.lua:39: in function 'forward'
train-cpu.lua:106: in function 'opfunc'
/home/sanoob/torch/install/share/lua/5.1/optim/sgd.lua:43: in function 'sgd'
train-cpu.lua:115: in function 'train'
train-cpu.lua:190: in main chunk
at line
local outputs = model:forward(inputs)
Should we declare inputs as FloatTensor too?