Solved by albanD in post #2 
              
              
                Hi, 
First you should never call the .forward() method of an nn.Module. But call the module instance with the inputs. 
The advantage of the class version are: 

For modules with parameters or buffers, it is nicer to just save these in self
It allows to use simple structure constructions like nn.Sequ…