会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
编程coding小白
博客园
首页
新随笔
联系
订阅
管理
2021年9月8日
Pytorch中保存模型
摘要: pytorch中模型的保存pth,pkl,pt格式 pytorch保存模型非常简单,主要有两种方法: 只保存参数;(官方推荐) 保存整个模型 (结构+参数)。 由于保存整个模型将耗费大量的存储,故官方推荐只保存参数,然后在建好模型的基础上加载。本文介绍两种方法,但只就第一种方法进行举例详解。 模型保
阅读全文
posted @ 2021-09-08 18:04 编程coding小白
阅读(940)
评论(0)
推荐(0)
编辑
公告