PyTorch中怎么加载本地权重文件_使用torch.load读取pth文件

老伟吖_9945

老伟吖_9945

2026-04-09

786人浏览

原创

最常见原因是相对路径错误,应使用绝对路径并检查文件存在性;文件损坏或格式不符会导致EOF或UnpicklingError;键名不匹配需核对并清洗前缀;跨设备加载必须指定map_location。

pytorch中怎么加载本地权重文件_使用torch.load读取pth文件

torch.load 读取 .pth 文件时路径不对或报错 OSError: No such file or directory

最常见的是相对路径写错,尤其在 Jupyter 或不同工作目录下运行时。torch.load 不会自动搜索项目根目录或模型目录,它只认当前 os.getcwd() 下的相对路径或绝对路径。

  • 用 os.path.abspath("model.pth") 或 Path(__file__).parent / "model.pth" 显式构造路径,别信“应该就在旁边”
  • 加载前先用 os.path.exists("model.pth") 检查,比报错后翻路径快得多
  • 如果文件在子目录(如 weights/ckpt.pth),别漏掉斜杠——"weightsckpt.pth" 是典型手误

加载时报 RuntimeError: unexpected EOF 或 pickle.UnpicklingError

说明文件损坏、不完整,或根本不是 PyTorch 保存的 .pth 文件(比如是 ONNX、TensorFlow checkpoint,或者下载中断)。

  • 用 file model.pth(Linux/macOS)或 Get-ChildItem model.pth | Select Length(PowerShell)看文件大小——空文件或几 KB 基本就是废的
  • torch.load 默认用 pickle 反序列化,不能加载用 torch.save(..., _use_new_zipfile_serialization=False) 旧方式保存的文件(极少见),但更可能是文件本身不合法
  • 别用文本编辑器强行打开或“修复” .pth 文件——它是二进制 ZIP 格式,改一个字节就全崩

加载后 state_dict 键名对不上,模型 load_state_dict 失败

权重文件里的键(如 "backbone.conv1.weight")和你模型实际参数名不一致,常见于:模型定义改动过、用了不同训练脚本、或权重来自 Hugging Face 等第三方库。

PyTorch Linux版 2.11.0
PyTorch Linux版 2.11.0

PyTorch 2.11.0 历史版本下载,来自 PyPI 官方发布,适合旧项目兼容、实验复现和指定环境安装。

下载
  • 先用 print(torch.load("model.pth").keys()) 看存的是啥(如果是 state_dict 直接存的),再用 print(list(model.named_parameters())) 对比
  • 键名差前缀?用 state_dict = {k.replace("module.", ""): v for k, v in state_dict.items()} 清洗(多卡训练保存的常见问题)
  • 键名顺序/结构完全不一样?别硬映射——确认你加载的是对应模型的权重,不是同名但不同架构的文件

用 map_location 避免 GPU 加载失败或显存泄漏

在 CPU 上加载原本存在 GPU 的 .pth 文件,或反过来,不指定 map_location 会直接 OOM 或报 Invalid device。

  • CPU 加载:必须写 torch.load("model.pth", map_location="cpu"),否则默认尝试加载到原设备(如 cuda:0)
  • GPU 加载但想指定卡:用 map_location="cuda:1",而不是靠 model.cuda() 后补——后者不改 state_dict 里张量的设备,load_state_dict 会失败
  • 安全写法:map_location=torch.device("cpu") 比字符串更明确,也兼容未来版本

PyTorch 的 torch.load 行为高度依赖文件来源和保存时的上下文,同一个 .pth 文件,在不同环境里可能因设备、路径、模型定义微小差异就加载失败——别假设“能跑通一次就永远没问题”。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

pytorch

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1831

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4904

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1789

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

29257

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3487

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3507

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1223

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

616

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2723

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程