我是在云GPU上训练的，训练完成之后，保存了最优模型字典pth文件，在到cpu本地加载的时候，我们需要读取pth文件，正确代码如下

     # 加载模型权重，确保映射到正确的设备model = models.resnet18(weights=models.ResNet18_Weights.IMAGENET1K_V1)  # 创建模型实例model=  torch.load('best-0.990.pth', map_location=device)model.to(device)  # 将模型映射到指定的设备

错误代码如下

import torchvision.models as models# 创建模型架构的实例
model = models.resnet18(pretrained=False)  # 或者其他您使用的模型架构# 加载模型的状态字典
model.load_state_dict(torch.load('best-0.990.pth', map_location=device), strict=False)

教程如下pytorch 状态字典:state_dict使用详解https://www.jb51.net/article/178720.htm