python2的torch model如何在python3中载入

最新推荐文章于 2025-07-20 00:21:40 发布
原创 最新推荐文章于 2025-07-20 00:21:40 发布 · 1k 阅读 · 1 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

博客主要围绕Python2的torch model在Python3中载入的问题展开。作者在载入https://github.com/ox-vgg/vgg_face2 vgg官方模型时遇到问题，该模型版本为Python2，而作者想在Python3环境下运行，经查询相关代码后成功载入。

python2的torch model如何在python3中载入

载入https://github.com/ox-vgg/vgg_face2 vgg官方模型时出了问题
该版本为Python2但我想在python3环境下跑
在查了源码和官方代码和一些网友提供的答案后，使用如下代码成功载入

import pickle
import io
from functools import partial
MainModel = imp.load_source('MainModel', 'resnet50_ft_pytorch.py')
pickle.load = partial(pickle.load, encoding="latin1")
pickle.Unpickler = partial(pickle.Unpickler, encoding="latin1")
param_dict = torch.load(model_path)
【RKNN】RKNN-Toolkit2 Python 模型加载API函数详解
浩瀚之水的专栏
05-02 324
RKNN-Toolkit2 提供多种函数加载不同框架的预训练模型，将其转换为 RKNN 格式。
【RKNN】RKNN-Toolkit2 Python API函数详解
浩瀚之水的专栏
04-29 555
RKNN-Toolkit2 的 API 设计以开发者效率为核心，覆盖模型转换、量化、部署全流程。通过灵活的参数配置（如多核调度、混合量化）和调试工具（性能分析、精度对比），可快速适配复杂边缘场景。
Pytorch中python3加载python2预训练模型
tiancity的博客
04-02 1837
最近在学习别人代码的时候发现环境要求python2.7，但是又懒得搭建anaconda虚拟环境，因此在python3，pytorch 0.4下进行了测试，遇到了一下问题 UnicodeDecodeError: 'ascii' codec can't decode byte 0xd0 in position 0: ordinal not in range(128) 查阅相关资...
Pytorch 笔记 -- model模型
mznfirst的博客
01-04 949
1. 保存载入模型 import torch torch.save(model,'model_name') #将网络模型、模型参数全部保存 torc.save(model.state_dict(),'model_name.pkl') #只保存模型参数 两种载入方式也不相同 torch.load('model_name') from torchvision import models mod...
Pytorch 里面model部分属性简析
张哥频道：电生理信号和磁共振成像
07-07 1836
Pytorch 里面model 部分属性简析
深入解析 PyTorch 的 torch.load() 函数：用法、参数与实际应用示例
m0_73640344的博客
11-28 2866
深入解析 PyTorch 的 torch.load() 函数：用法、参数与实际应用示例函数是一个在PyTorch中用于加载通过保存的序列化对象的核心功能。这个函数广泛应用于加载预训练模型、模型的状态字典（state dictionaries）、优化器状态以及其他PyTorch对象。它利用Python的反序列化能力，特别地对张量的底层存储（storages）进行了特殊处理，以支持跨设备加载和内存效率。
Pytorch之torch.load()语法、参数和实际应用案例
最新发布
王国平的博客
07-20 977
`torch.load()` 是 PyTorch 中用于加载保存在磁盘上的对象（如模型、张量、字典等）的核心函数。它支持从文件或类文件对象中反序列化数据，是模型保存与恢复的关键工具。
python版tensorrt推理
Thomas_Cai的记忆殿堂
01-25 4688
windows10/11场景下，在python中进行tensorrt推理加速实践，模型推理加速必备~
Python × CARLA：如何在自动驾驶仿真世界里打造智能驾驶系统？
Echo_Wish
05-14 819
在人工智能与自动驾驶的浪潮中，真实世界的测试成本高昂，而**自动驾驶仿真**已成为开发者训练和测试 AI 驾驶算法的关键技术手段。其中，**CARLA**（Car Learning to Act）作为**开源自动驾驶仿真平台**，凭借其真实感强、高度可定制、可集成深度学习框架等特性，成为**自动驾驶算法开发者的最优选择**。
torch载入模型
03-25
此方法的前提条件是在保存阶段调用了 `torch.save(model.state_dict(), PATH)` 来存储模型参数[^2]。 --- #### 跨设备迁移模型 当需要将模型从 GPU 迁移到 CPU 或反之亦然时，可以通过指定映射位置来完成这一过程...
pytorch(一)模型加载函数torch.load()
热门推荐
凝眸伏笔的博客
11-08 11万+
1.作用：用来加载torch.save()保存的模型文件。 torch.load()先在CPU上加载，不会依赖于保存模型的设备。如果加载失败，可能是因为没有包含某些设备，比如你在gpu上训练保存的模型，而在cpu上加载，可能会报错，此时，需要使用map_location来将存储动态重新映射到可选设备上，比如map_location=torch.device('cpu')，意思是映射到cpu上，在cpu上加载模型，无论你这个模型从哪里训练保存的。 一句话：map_location适用于修改模型能在gpu.
torch.load()
weixin_48697962的博客
07-26 7504
torch.load()
【PyTorch】基础学习：一文详细介绍 torch.load() 的用法和应用
创作高质量博文，分享知识，共同进步！
03-17 8174
🔥掌握PyTorch核心技能，一文读懂torch.load()的奥秘！💡 🌟深度解读torch.load()基本概念，让你快速上手！📚 🚀探索torch.load()高级用法，轻松应对复杂场景！💡 🔄torch.load()与torch.save()的完美配合，实现模型的无缝迁移！🔄 🔍常见问题一网打尽，让你避免踩坑！💣 🌐torch.load()在实际项目中大放异彩，助你一臂之力！🚀 💼总结与展望，让你站在巨人的肩膀上，看得更远！🔭 #PyTorch学习 #torch.loa
torch中的model.eval()、model.train()详解
you 是 mine
05-23 3453
torch中的model.eval()、model.train()详解
pytorch之torch.save()和torch.load()方法详细说明
yueguang8的博客
03-21 4527
torch.save()和torch.load()是PyTorch中用于模型保存和加载的函数。它们提供了一种方便的方式来保存和恢复模型的状态、结构和参数。可以使用它们来保存和加载整个模型或其他任意的Python对象，并且可以在加载模型时指定目标设备。
torch.load() 、torch.load_state_dict() 详解
you 是 mine
03-08 2万+
torch.load() 、torch.load_state_dict() 详解
torch.save 和 torch.load函数介绍
qq_27390023的博客
10-25 1447
torch.save：用于保存对象到文件，支持张量、字典、模型和优化器的state_dict等。torch.load：用于加载保存的文件内容，并支持在不同设备之间加载。
Python3 Django的Model详解
Luzhuo 的博客
03-12 3473
Python3 Django的Model详解 本文由 Luzhuo 编写,转发请保留该信息. 原文: http://blog.youkuaiyun.com/Rozol/article/details/79526259 以下代码以Python3.6.1为例 Less is more! Python 3.6.1 Django 1.8.2 (注: 由于2.0.2不支持mys...
1.torch.load()函数介绍
clhmliu的博客
11-17 4137
官网翻译过来
Python中查看.pth文件内容的方法详解
在Python中，尤其是深度学习和机器学习领域，`.pth`文件是一种非常常见且重要的文件格式。这类文件通常由PyTorch框架生成，用于保存训练好的神经网络模型的权重参数（state_dict）、整个模型结构、优化器状态或其他...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值