Solved by ptrblck in post #2 
              
              
                net.load_state_dict expects the actual state_dict as its input while you are passing your custom dict to it. 
Try to use: 
net.load_state_dict(torch.load(PATH)['state_dict'])

and it should work.