怎样在Python中使用PyTorch的torch.hub加载预训练模型

千静酱_7862

千静酱_7862

2026-10-01

282人浏览

原创

torch.hub.load加载失败主因是网络无法访问github、仓库无hubconf.py或入口函数、pytorch版本过低(

怎样在python中使用pytorch的torch.hub加载预训练模型

torch.hub.load 为什么加载失败或报错

直接调用 torch.hub.load 却提示 ModuleNotFoundError、HTTPError 404 或卡在下载,通常是因为仓库地址、模型名拼写、PyTorch 版本不匹配,或者网络无法访问 GitHub(尤其国内用户)。PyTorch Hub 依赖 GitHub 仓库的 hubconf.py 文件,该文件必须存在且导出合法的 entry_point 函数。

  • 确保 PyTorch ≥ 1.6(旧版本 hub 支持有限)
  • 检查目标仓库是否仍在维护:例如 pytorch/vision 是官方支持的,但第三方仓库如 facebookresearch/detr 需确认其 hubconf.py 是否更新适配当前 PyTorch
  • 国内用户建议加 source='github' 并配置代理,或改用镜像源(见下条)

如何指定 GitHub 分支或本地缓存路径

torch.hub.load 默认从 master 分支加载,但很多项目已迁移到 main;同时默认缓存到 ~/.cache/torch/hub/,若磁盘空间不足或想复用已有模型,需手动控制。

  • 指定分支:传入 git_tag='main'(不是 branch=)参数,例如:
    model = torch.hub.load('pytorch/vision', 'resnet18', pretrained=True, git_tag='main')
  • 指定缓存目录:设置环境变量 TORCH_HOME,或调用前执行 torch.hub.set_dir('/path/to/my/hub_cache')
  • 跳过检查强制重载:加 force_reload=True(仅调试用,会清空对应仓库缓存)

pretrained=True 和 weights 参数的区别(PyTorch ≥ 1.12)

从 PyTorch 1.12 开始,pretrained=True 已被弃用,新接口统一使用 weights=xxx,类型是具体权重类(如 ResNet18_Weights.IMAGENET1K_V1),而非布尔值。混用会导致 TypeError 或静默降级为随机初始化。

python-code-analyz
python-code-analyz

专业Python代码分析与优化,支持语法检查、安全扫描、性能评估、复杂度分析及重构后优化代码生成。

下载
  • 正确写法(推荐):
    from torchvision.models import resnet18, ResNet18_Weights<br>model = resnet18(weights=ResNet18_Weights.IMAGENET1K_V1)
  • 若坚持用 torch.hub,仍可传 pretrained=True,但它底层会自动映射为最新权重常量——前提是 hubconf.py 已更新(很多老仓库没做)
  • 自定义权重路径不被 torch.hub.load 原生支持,需加载后手动 load_state_dict

加载后如何验证模型结构和输入输出

Hub 加载的模型未必开箱即用:可能缺少预处理逻辑、归一化参数不对、或返回格式与文档不符(比如 DETR 返回字典而非张量)。不能只看是否不报错。

  • 打印模型结构:print(model),重点关注最后一层是否匹配你的任务(如分类数)
  • 模拟输入测试前向传播:
    x = torch.randn(1, 3, 224, 224)<br>out = model(x)<br>print(out.shape)  # 确认输出维度符合预期
  • 检查是否含预处理:部分 Hub 模型(如 segmentation/deeplabv3_resnet101)在 hubconf.py 中封装了 transforms,需调用 model.transforms() 获取,否则自己得补 Normalize

实际项目中,torch.hub 更适合快速原型验证;生产环境建议直接用 torchvision.models 或明确 clone 仓库 + git checkout 版本,避免隐式依赖远程分支变更。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

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

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

下载

相关标签:

python pytorch

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

相关专题

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

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

2023.07.20

1631

4

python能做什么
python能做什么

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

2023.07.25

3984

7

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

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

2023.07.31

1629

3

python教程
python教程

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

2023.08.03

22977

23

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

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

2023.08.04

2827

5

python eval
python eval

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

2023.08.04

2867

5

scratch和python区别
scratch和python区别

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

2023.08.11

1123

5

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

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

2023.08.10

596

4

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

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

2023.08.11

2223

5

热门下载

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

精品课程

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