distro-cl
distro-cl copied to clipboard
Toch-cl not working with CycleGAN?
Hi hughperkins, thanks for the great job!
I've used Torch-cl on several neuro networks, unfortunately couldn't even load it on CycleGAN.
I've installed all necessary dependancies (including nn graph), but still receive error (below). Seems like nn is not recognised. With usual Torch and same packages installed everything is fine, but CPU mode only... Any way to fix it?
/home/vitek/torch-cl/install/bin/luajit: /home/vitek/torch-cl/install/share/lua/5.1/torch/File.lua:334: unknown Torch class <nn.SpatialReflectionPadding> stack traceback: [C]: in function 'error' /home/vitek/torch-cl/install/share/lua/5.1/torch/File.lua:334: in function 'readObject' /home/vitek/torch-cl/install/share/lua/5.1/torch/File.lua:360: in function 'readObject' /home/vitek/torch-cl/install/share/lua/5.1/torch/File.lua:360: in function 'readObject' /home/vitek/torch-cl/install/share/lua/5.1/nn/Module.lua:154: in function 'read' /home/vitek/torch-cl/install/share/lua/5.1/torch/File.lua:342: in function 'readObject' /home/vitek/torch-cl/install/share/lua/5.1/torch/File.lua:391: in function 'load' /home/vitek/CycleGAN/util/util.lua:192: in function 'load_test_model' ./models/one_direction_test_model.lua:24: in function 'Initialize' test.lua:79: in main chunk [C]: in function 'dofile' ...k/torch-cl/install/lib/luarocks/rocks/trepl/scm-1/bin/th:145: in main chunk [C]: at 0x00405e90