Solved by ptrblck in post #2 
              
              
                The map_location argument remaps the tensors to the specified device. This is useful, if your current machine does not have a GPU and you are trying to load a tensor, which was saved as a CUDATensor.

From the docs: 

 torch.load()  uses Python’s unpickling facilities but treats storages, which und…