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
投稿猜你喜欢
- 目录del：根据索引值删除元素pop()：根据索引值删除元素remove()：根据元素值进行删除clear()：删除列表所有元素在 Pyth
有一个优秀的库可以使用————demjson示范链接http- 1.collatz序列编写一个名为 collatz()的函数，它 有一个名为 number 的参数。如果参数是偶数， 那么 collatz(
- 在这个教材中,我们假定你已经安装了Scrapy。假如你没有安装,你可以参考这个安装指南。我们将会用开放目录项目(dmoz)作为我
- PIL图片如何按比例裁剪问题描述如图片比例为 1:1 裁剪为 4:31.jpg解决方案from PIL import Imagedef im
- 【目标需求】查看某一个字符串是否为中文字符串【解决办法】def check_contain_chinese(check_str): for
- 这篇文章主要介绍了python如果快速判断数字奇数偶数,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋
- 一、常用按键按键说明Keys.BACK_SPACE回退键（BackSpace）Keys.TAB制表键（Tab）Keys.ENTER回车键（E
- 1.定义ResNet残差模块一个block中，有两个卷积层，之后的输出还要和输入进行相加。因此一个block的前向流程如下：输入x→卷积层→
- 网页的圆角处理，其实最开始的九宫格的表格处理是挺方便的，只是现在都不用表格布局了，自然就被非法取缔了。呵呵。微软的有VML画圆角，由于只是它
- flask响应错误处理及errorhandler应用@app.errorhandler(404)def page_not_found(err
- 1. OpenCV：模板匹配。 获得小跳棋中心位置2.
- 看代码吧~class ConvNet(nn.module): def __init__(self, num_cla
- adfuller函数返回值的参数说明from statsmodels.tsa.stattools import adfullert = ad
- 本文实例讲述了Python将名称映射到序列元素中的方法。分享给大家供大家参考，具体如下：问题：希望通过名称来访问元素，减少结构中对位置的依赖
- 这篇文章主要介绍了Python搭建HTTP服务过程图解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋
- MySQL有多种存储引擎：MyISAM、InnoDB、MERGE、MEMORY(HEAP)、BDB(BerkeleyDB)、EXAMPLE、
- 一、框架菜单1.1 common模块1.2 其他二、Excel接口测试案例编写三、读取Excel测试封装（核心封装）excel_utils.
- 备注：Ken Henderson 从开发者的角度来阐述了SQL SERVER 2000内存管理的内部机制简介在本专栏中，我们将从一个开发者的
- 如下所示：# coding = utf-8import requestsimport jsonhost = "http://47.
