报错：RuntimeError: Attempted to set the storage of a tensor on device “cuda:0“ to a storage on differe

用u-net的两个骨干网时，一个vgg，一个resnet50，换到resnet50时一直报这个错误，找了很多资料都没有解决。

pretrained_dict = torch.load(model_path, map_location=device)

一直显示这行报错，后来突发奇想把map_location这个参数删除了，结果能运行成功了。