pytorch模型存储的2种实现方法
作者：慢行厚积 发布时间：2023-10-06 11:37:24
标签：pytorch,模型,存储
1、保存整个网络结构信息和模型参数信息：
torch.save(model_object, './model.pth')
直接加载即可使用：
model = torch.load('./model.pth')
2、只保存网络的模型参数-推荐使用
torch.save(model_object.state_dict(), './params.pth')
加载则要先从本地网络模块导入网络，然后再加载参数：
from models import AgeModel
model = AgeModel()
model.load_state_dict(torch.load('./params.pth'))
来源：https://www.cnblogs.com/wanghui-garcia/p/11236382.html
0
投稿猜你喜欢
- 1.首先，我们需要过滤所有客户端提交的内容,其中包括?id=N一类,另外还有提交的html代码中的操作数据库的select及asp文件操作语
- 1. 域名查询万网提供了域名查询接口，接口采用HTTP协议：接口URL：http://panda.www.net.cn/cgi-bin/ch
前言这是俺写的第一篇关于python的博客，分享一下我所整合的python中的随机数函数，就当作是自己的一个笔记，也可以供像我一样的pyth- 准备开始学习Python，但是刚准备环境搭建时就遇到了下面的错误：仔细的看了看，说是缺少DLL。对于这个问题的解决办法：方法一：1. 在安装
- 一、源码安装首先安装开发工具包yum groupinstall -y "Development tools"安装依赖软件
- 安装使用pip install XlsxWriter来安装,Xlsxwriter用来创建excel表格,功能很强大,下面具体介绍:1.简单使
- 本文实例讲述了python实现自动登录人人网并访问最近来访者的方法，分享给大家供大家参考。具体方法如下：##-*- coding : gbk
- 选中需要操作的字符Ctrl + R 替换Ctrl + Shift + F 全局查找Ctrl + Shift + R 全局替换来源：http:
- 先把这个script加到你的页面里:http://code.google.com/p/doufu/source/browse/trunk/n
- SeaTunnel是什么？SeaTunnel下一代高性能、分布式、海量数据集成框架。SeaTunnel是一个非常易于使用的超高性能分布式数据
- 一、map 1.基本介绍map 是 key-value 数据结构，又称为字段或者关联数组。类似其它编程语言的集合， 在编程中是经常
- 关于webpack的配置和使用，网上已经有许多文章了，大多是在讲单页应用，当我们需要打包多个html时，事情就变得麻烦起来。怎么在webpa
- 一、引言池化技术在Java中应用的很广泛，简而论之，使用对象池存储某个实例数受限制的实例，开发者从对象池中获取实例，使用完之后再换回对象池，
- 本文实例总结了MySQL数据库优化技术的索引用法。分享给大家供大家参考，具体如下：这里紧接上一篇《MySQL数据库优化技术之配置技巧总结》，
- 刚才显示数据的时候遇到一个日期里面带T的问题，就是天数跟小时数之间出现了一个T。 表字段里面也没有这个T，后来查询度娘，是因为json处理的
- 前言bufio包实现了带缓冲的I/O，它封装了io.Reader和io.Writer对象，然后创建了另外一种对象（Reader或Writer
- 列表更多的方法index()：返回指定数据所在位置的下标 (注意：如果查找的数据不存在则报错。)。count()：统计指定数据在当前列表中出
- 一键执行虚拟机一键安装python3.8环境，只需将网络适配器改为nat模式即可（确保主机能够上网），随后将tar包放入/root目录下，执
- 本文实例讲述了Python实现简单的多任务mysql转xml的方法。分享给大家供大家参考，具体如下：为了需求导出的格式尽量和navicat导
- 当后端并未编写后台json数据，但是前端又要使用数据进行页面的编写，这是便可用mockjs编写一个假的数据源，发送ajax请求并获取数据1.
