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中使用Matplotlib进行多图绘制的详细教程
Matplotlib是Python中强大的数据可视化工具,支持绘制多图,这篇文章主要介绍了Python中使用Matplotlib进行多图绘制的相关资料,包括使用subplot、subplots和GridSpec进行布局,并提供了调整图表样式和布局的方法,需要的朋友可以参考下2024-11-11
在python中我们偶尔会用到输出不换行的效果，python2中使用逗号，即可，而python3中使用end=''来实现的，这里简单为大家介绍一下,需要的朋友可以参考下2017-11-11
本文主要介绍了python里面单双下划线的区别详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-04-04
本文将详细介绍如何使用 Python 调用 DeepSeek API,实现流式对话并保存对话记录,相比 Go 版本,Python 实现更加简洁优雅,适合快速开发和原型验证,文中通过代码示例讲解的非常详细,需要的朋友可以参考下2025-02-02
这篇文章主要介绍了Python subprocess库的使用详解，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-10-10
装饰器的运用是Python编程中的一项高级技巧,这里我们由浅入深,整理了12步入门Python中的decorator装饰器使用方法,需要的朋友可以参考下2016-06-06
大家好，本篇文章主要讲的是Python进制转换用法详解，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下2022-01-01
这篇文章主要介绍了Python中的内置函数isdigit()，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-11-11
在Python中,实现定时任务是一个常见的需求,无论是在自动化脚本、数据处理、系统监控还是其他许多应用场景中,Python提供了多种方法来实现定时任务,包括使用标准库、第三方库以及系统级别的工具,本文将详细介绍几种常见的Python定时任务实现方式2024-08-08
这篇文章主要为大家介绍了python调用matlab，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2021-11-11
最新评论