unknown Torch class <nn.gModule>
TheTesla opened this issue · comments
Stefan Helmert commented
steps to reproduce:
import PyTorchAug x = PyTorchAug.load('umich-stacked-hourglass.t7')
output:
~/.luarocks/share/lua/5.1/torch/File.lua:343: unknown Torch class <nn.gModule>
but there is a ~/.luarocks/share/lua/5.1/nngraph/gmodule.lua
also there is a ~/.luarocks/share/lua/5.1/nn directory - maybe something incompatible/confusing/wrong require?
How to solve the problem?
Stefan Helmert commented
I found the misstake, I have to:
import PyTorch PyTorch.require('nngraph')
explicitely