加载模型时收到错误

作者: pawara101创建于 2026年7月23日更新于 2026年7月23日

不错的工作,而我检查你的代码 发现一个问题 在“tourios/01-basics/pytorch basics/main.py”

-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
不挑剔的 Error 回溯( 最近的一次呼叫)
/tmp/ipy内核 58/1496455266.py 在 <细胞行:0> ()
5#保存并装入整个型号.
6 火炬.save(resnet,"model.ckpt") 互联网档案馆的存檔,存档日期2013-12-21.
----> 7 型号 = 火炬. load ('model.ckpt')
第8条
9 # 只保存并加载模型参数(建议).

/usr/local/lib/python3.12/dist-packages/torch/serialization.py in load(f,地圖-地點取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取取
页:1
1547年除 pick菜. 摘取错误为 e:
- > 1548 起泡菜. unpickling Error ( get wo  message (str(e))) from None
1549 返回   载( )
1550 打开 zipfile,

不挑剔错误: 重量只负载失败 。 这个文件仍然可以装入,为了做到这样,你有两个选项,只有当您信任检查点的来源时才能采取这些步骤.
(1) 在PyTorch 2.6中,我们把`torch.load'中`quenters  only'参数的默认值从`false'改为`True'。 重新运行`torch.load',将`重量'设为`false',可能成功,但可能导致任意执行代码。 只有在您从可靠来源得到文件时才会这么做 。
(2) 或者,要装入“重量 只=True”,请检查以下错误信息中建议的步骤。
WightsUnpickler出错:不支持全局:Global火炬视觉. models.resnet. ResNet 默认不是一个允许的全球 。 请使用“ torch.serialization.add safe globals([torchvision.models.resnet.ResNet]) ” 或“ torch.serialization.safe globals ([torchvision.models.resnet.ResNet]])” 上下文管理器, 如果您信任此类/功能, 则允许此全局列表 。

请检查pytorch.load的文档, 以便更多了解默认接受的类型, 仅使用权重 https://pytorch.org/docs/stable/create/torch.load.html。

我认为它将迅速解决以下问题:

zz
{\fn方正粗倩简体\fs12\an8\1cHFFFF00\b0}#############################################################################################################################################################################################################################################
# 7. 保存并装入模型 #
{\fn方正粗倩简体\fs12\an8\1cHFFFF00\b0}#############################################################################################################################################################################################################################################

# 保存并装入整个模型。
火炬. save(resnet,"model.ckpt") (中文(简体) ).
model=火炬.load ("model.ckpt",重量 只为虚构)

# 只保存并加载模型参数(建议).
火炬.save(resnet.state dict (), “params.ckpt () ”
resnet.load state dict(torch.load ('params.ckpt')) (英语).

内容来源: yunjey/pytorch-tutorial