Solved by tom in post #2 
              
              
                This 
self.model = nn.Module

doesn’t really look like you’re supposed to do it. You could use nn.Module(). 
An even cleaner way might be to use a ModuleList or Sequential. 
Best regards 
Thomas