About test

[xurong1981]

when I use "CUDA_VISIBLE_DEVICES=0 th test.lua -img ./imgs/comic_input.bmp -output ./output.bmp -model ./models/SRResNet_MSE_100.t7" to make a test, it is no problem.

However, when I changed test image by my own data (such as a jpg image of 640*480), and it gave an error of "out of memory". Someone can help resolve the following problem.

---

THCudaCheck FAIL file=/tmp/luarocks_cutorch-scm-1-3199/cutorch/lib/THC/generic/THCStorage.cu line=66 error=2 : out of memory
~/torch/install/bin/luajit: ~/torch/install/share/lua/5.1/nn/Container.lua:67:
In 8 module of nn.Sequential:
~/torch/install/share/lua/5.1/nn/PixelShuffle.lua:55: cuda runtime error (2) : out of memory at /tmp/luarocks_cutorch-scm-1-3199/cutorch/lib/THC/generic/THCStorage.cu:66
stack traceback:
[C]: in function 'resize'
/torch/install/share/lua/5.1/nn/PixelShuffle.lua:55: in function </torch/install/share/lua/5.1/nn/PixelShuffle.lua:17>
[C]: in function 'xpcall'
~/torch/install/share/lua/5.1/nn/Container.lua:63: in function 'rethrowErrors'
~/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'
test.lua:41: in function 'main'
test.lua:46: in main chunk
[C]: in function 'dofile'
~/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: 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'
~/torch/install/share/lua/5.1/nn/Container.lua:67: in function 'rethrowErrors'
~/torch/install/share/lua/5.1/nn/Sequential.lua:44: in function 'forward'
test.lua:41: in function 'main'
test.lua:46: in main chunk
[C]: in function 'dofile'
~/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk
[C]: at 0x00405d50

---

Thank you for your help.