You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

加载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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.04.14 16:39:31