- 博客(1)
- 收藏
- 关注
转载 【无标题】
pytorch保存模型等相关参数,利用torch.save(),以及读取保存之后的文件本文分为两部分,第一部分讲如何保存模型参数,优化器参数等等,第二部分则讲如何读取。假设网络为model = Net(), optimizer = optim.Adam(model.parameters(), lr=args.lr), 假设在某个epoch,我们要保存模型参数,优化器参数以及epoch一、先建立一个字典,保存三个参数:state = {‘net’:model.state_dict(), ‘opt
2022-04-29 20:42:43
93
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人