pytorch模型存储的2种实现方法
更新时间：2020年02月14日 12:00:56 作者：慢行厚积
今天小编就为大家分享一篇pytorch模型存储的2种实现方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
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'))
以上这篇pytorch模型存储的2种实现方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
中文分词是中文文本处理的一个基础性工作，然而长久以来，在Python编程领域，一直缺少高准确率、高效率的分词组件。下面这篇文章主要给大家介绍了关于python中文分词教程之前向最大正向匹配算法的相关资料，需要的朋友可以参考下。2017-11-11![python chinesecalendar报错:"no available data for year {}, only year between [{}, {}] supported"](//img.jbzj.com/images/xgimg/bcimg1.png)
python chinesecalendar报错:"no available data for ye
这篇文章主要介绍了python chinesecalendar报错：“no available data for year {}, only year between [{}, {}] supported“的相关知识,需要的朋友可以参考下2023-03-03
正则表达式有元字符及不同组合来构成，通过巧妙的构造正则表达式可以匹配任意字符串，并完成复杂的字符串处理任务，希望本片文章能给你带来帮助2021-09-09
这篇文章主要介绍了浅谈pytorch中的dropout的概率p，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-05-05
本篇文章主要介绍了python3 pillow生成简单验证码图片的示例，非常具有实用价值，需要的朋友可以参考下2017-09-09
python之plt.hist函数的输入参数和返回值的用法解释
这篇文章主要介绍了python之plt.hist函数的输入参数和返回值的用法解释,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-10-10
每个人的电脑里都会有不想让别人知道的隐私，或者是上班时间偷偷摸鱼怕被发现的小秘密。那怎么办？就干脆把隐私锁起来！从源头上杜绝被他人偷窥自己的隐私。本文就来用Python实现一个windows倒计时锁屏功能，需要的可以参考一下2023-04-04
本文详细介绍了Playwright的常用操作方法，包括获取页面元素、点击按钮和链接等。这些方法可以帮助开发者更加高效地进行自动化测试和爬虫开发,需要的可以参考一下2023-05-05
这篇文章主要为大家介绍了Python NumPy科学计算库的高级应用深入详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-07-07
这篇文章主要介绍了Django Form设置文本框为readonly操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-07-07
![python chinesecalendar报错:"no available data for year {}, only year between [{}, {}] supported"](http://img.jbzj.com/images/xgimg/bcimg1.png)
最新评论