热门标签
热门文章
- 1【芯片设计- RTL 数字逻辑设计入门 5 -- RTL 全加器实现及验证】_简单的rtl设计
- 2关系型数据库和非关系型数据库
- 3docker swarm 常用命令简介以及使用案例_swarm集群 指令
- 4vscode远程连接服务器报错_vscode转发端口报错
- 5IPsec 图解指南_ipsec 相关rfc
- 6程序员可以做什么副业？我的经历告诉你_程序员可以做那些副业
- 72024新版SonarQube+JenKins+Github联动代码扫描（1）-JenKins安装与配置_sonarqube + jenkins
- 8基于Java+SpringBoot+Vue+uniapp微信小程序零食商城系统设计和实现_springboot商城实战项目h5微信小程序java源码_商城微信小程序springboot源码下载
- 920211030 1+X 中级实操考试（id：2498）_objectutil
goodsutil 查询所有商品中价钱最贵的商品名称 - 10MVC拦截器、ThreadLocal来进行登录拦截
当前位置: article > 正文
如何保存与调用 pytorch 训练好的网络模型_torch.save 直接将整个网络进行了保存,可以直接调用吗
作者：一键难忘520 | 2024-07-13 07:40:30
赞
踩
torch.save 直接将整个网络进行了保存,可以直接调用吗
如何保存与调用 pytorch 训练好的网络模型_TheWindOfJune的博客-CSDN博客_pytorch训练好的模型保存和使用问题描述深度神经网络的模型训练起来十分困难，那么训练好的 pytorch 网络模型就应该把它保存起来，下次使用的时候直接调用就行了，应该怎么保存这个模型呢？解决方法Pytorch 主要提供了两种方法，分别是：保存模型参数的方法 和 保存整个模型的方法方法一：只保存模型参数#保存torch.save(the_model.state_dict(), PATH)#读取the_model = TheModelClass(*args, **kwargs)the_model.load_sthttps://blog.csdn.net/weixin_43450646/article/details/106931575问题描述
深度神经网络的模型训练起来十分困难，那么训练好的 pytorch 网络模型就应该把它保存起来，下次使用的时候直接调用就行了，应该怎么保存这个模型呢？
解决方法
Pytorch 主要提供了两种方法，分别是：保存模型参数的方法 和 保存整个模型的方法
方法一：只保存模型参数
- #保存
- torch.save(the_model.state_dict(), PATH)
- #读取
- the_model = TheModelClass(*args, **kwargs)
- the_model.load_state_dict(torch.load(PATH))
方法二：保存整个模型
- #保存
- torch.save(the_model, PATH)
- #读取
- the_model = torch.load(PATH)
还可以参看这篇文章：
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/一键难忘520/article/detail/818314
推荐阅读
相关标签
