Solved by justusschock in post #7 
              
              
                Your Problem lies within your saving/loading code: 
You save your model via torch.save(net, save_path) where net is an instance of nn.DataParallel 
I would recommend to change a few things: 


You should only save the state dict instead of your model. See this post for further details 


You should …