PyTorch笔记: GPU上训练的模型加载到CPU/错误处理Attempting to deserialize object on a CUDA device but torch.cuda.is_a
我之前在GPU上训练了一个模型,同时把模型的参数记录在resnet18_ultimate.pkl上在本地的CPU上,我想把参数加载,于是一开始我是这么写代码的:importtorchimporttimmresnet=timm.create_model('resnet18',num_classes=1000,pretrained=True)model=torch.nn.Sequential(resn