Solved by ptrblck in post #5 
              
              
                I still think that you are hunting down two separate issues. 
torch.load(..., map_location) will load the tensor or state_dict onto the specified device. 
However, since you are piping this operation directly to model.load_state_dict, internally most likely param.copy_ will be used, which will then …