不同版本pytorch训的模型之间的转化

悠悠 2022-10-22 13:55 314阅读 0赞

写在开头: 本文仅记录自己遇到的问题，方便之后查阅。

最近用不同版本的pytorch训了一些模型，但使用pytorch1.3加载pytorch1.6训练的模型失败，显示如下所示等一些乱七八糟的错误。

![20210317163611114.png][]

但是没关系，因为本宝想到了一个好的解决办法，hiahiahia~

1. pytorch1.6 加载它自身生成的模型是没问题的，然后把加载后的模型dict生成pickle文件, 代码如下:

import pickle as pkl

import torch

info\_dict = torch.load('model\_final.pth')

with open('pkl\_model\_final.pth', 'wb') as f:

pkl.dump(info\_dict, f)

2 pytorch1.3 加载时，使用以下命令:

model = YOUR\_MODEL(args)

with open( '/pkl\_model\_final\_pth', 'rb') as f:  
        info\_dict = pkl.load(f)

model.load\_state\_dict(info\_dict\['model'\])

这样就可以保证不同pytorch版本之间可以互相加载彼此训练的模型了，简直完美，棒棒哒!

[20210317163611114.png]: /images/20221022/a4b65716715040a28e202e30d9a825b0.png

发表评论取消回复

表情：

评论列表（有 0 条评论，314人围观）

还没有评论，来说两句吧...

相关阅读

相关微软 Windows Server 版本对比：了解不同版本之间的差异

类似于客户端 Windows 版本，Windows Server 也分别提供了各种版本。 Windows Server 是一个操作系统，旨在管理和控制它周围的网络基础架构。每个

Bertha 。/ 2024年03月17日 15:44/ 0 赞/ 93 阅读

相关 Pytorch部署方案（二）：利用LibTorch（C++库）来调用训好的模型【trace跟踪PyTorch训练好的模型--＞用Libtorc调用PyTorch训练好的模型】

利用C++来调用PyTorch训练好的模型，其实也是利用官方强大的LibTorch库。一、LibTorch的安装虽然说安装，其实就是下载官方的LibTorch包而已

ゞ浴缸里的玫瑰/ 2023年09月29日 13:44/ 0 赞/ 48 阅读

相关 Pytorch预训练模型文件转化为onnx格式

代码如下 coding: utf-8 import torch from model.wacnet import WACNet 导入自己构建好的模型

偏执的太偏执、/ 2022年12月11日 12:24/ 0 赞/ 333 阅读

相关不同版本pytorch训的模型之间的转化

写在开头: 本文仅记录自己遇到的问题，方便之后查阅。最近用不同版本的pytorch训了一些模型，但使用pytorch1.3加载pytorch1.6训练的模型失败，显示

悠悠/ 2022年10月22日 13:55/ 0 赞/ 315 阅读

相关不同版本PHP之间cURL的区别

PHP的cURL支持通过给CURL\_POSTFIELDS传递关联数组（而不是字符串）来生成multipart/form-data的POST请求。

Dear 丶/ 2022年08月23日 08:29/ 0 赞/ 269 阅读

相关不同版本服务器之间的区别与联系

不同版本服务器之间的区别与联系 1.Tomcat Tomcat 服务器是一个免费的开放源代码的Web 应用服务器，它是Apache 软件基金（Apache Soft

谁践踏了优雅/ 2022年08月08日 15:57/ 0 赞/ 310 阅读

相关 BYTE CString之间的转化

在VC中,CSTRING类的用法形式多样,丰富多彩.但有时与传统数据类型之间的转换成为了很多人的难题. 下面我来简单介绍它们之间的关系. 1 如何实现cstrin

妖狐艹你老母/ 2022年07月03日 12:57/ 0 赞/ 366 阅读

相关网络-HTTP不同版本之间的对比

\\\\不管在找实习还是找工作的时候，关于http协议以及延伸出的一些问题都会经常被问到。专门小结一下。 -------------------- \\\\首先介绍下HTT

布满荆棘的人生/ 2022年05月28日 02:20/ 0 赞/ 243 阅读

相关温度之间的转化

输入华氏温度f，输出对应的摄氏温度c，保留三位小数。提示：c=5(f-32)/9. include <STDIO.H> int main() {

电玩女神/ 2022年05月17日 04:28/ 0 赞/ 229 阅读

相关 tensorflow tfserving 部署多个模型、使用不同版本的模型

本篇主要介绍使用tfserving和docker同时部署多个模型，使用不同版本的模型，基本的流程与部署单个模型的过程类似，(关于运行tfserving容器使用单个模型进行预测的

梦里梦外;/ 2022年03月26日 10:53/ 0 赞/ 787 阅读