hughperkins / pytorch

Python wrappers for torch and lua

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

unknown Torch class <nn.gModule>

TheTesla opened this issue · comments

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?

I found the misstake, I have to:
import PyTorch PyTorch.require('nngraph')
explicitely