会员
周边
捐助
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
FishL
博观而约取,厚积而薄发
博客园
首页
新随笔
联系
订阅
管理
2023年7月31日
模型权重保存、加载、冻结(pytorch)
摘要: 1. 保存整个网络 torch.save(net, PATH) model = torch.load(PATH) 2. 保存网络中的参数(速度快,占空间小) torch.save(net.state_dict(),PATH) model_dict = model.load_state_dict(to
阅读全文
posted @ 2023-07-31 21:35 湾仔码农
阅读(1904)
评论(0)
推荐(0)
编辑
公告