加载PyTorch模型权重时触发_pickle.UnpicklingError错误的求助
加载PyTorch模型权重时触发_pickle.UnpicklingError错误的求助
我在尝试加载PyTorch模型权重时遇到了这个错误:
_pickle.UnpicklingError: invalid load key, '\x1f'.
下面是我加载权重的代码:
import os import torch import numpy as np # from data_loader import VideoDataset import timm device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') print('Device being used:', device) mname = os.path.join('./CDF2_0.pth') checkpoints = torch.load(mname, map_location=device) print("Checkpoint loaded successfully.") model = timm.create_model('legacy_xception', pretrained=True, num_classes=2).to(device) model.load_state_dict(checkpoints['state_dict']) model.eval()
我已经尝试过更换不同的PyTorch版本,还试着把权重文件的扩展名改成.zip,用压缩管理器打开来检查文件,但都没能解决问题。这些模型我前后训练了大概一个月,一共有40个左右,现在加载不了真的很急,希望能得到大家的帮助!
备注:内容来源于stack exchange,提问作者DevLoverUmar
相关产品推荐
相关产品推荐

