Solved by ptrblck in post #2 
              
              
                Plain Python containers, such as list and dict won’t be properly registered inside your module, so use nn.ModuleDict in that case (or nn.ModuleList instead of list).