torch.save()用法

爱学习的人工智障

于 2019-09-27 14:55:54 发布

阅读量4.5w

点赞数 9

分类专栏： pytorch

版权

88 篇文章

订阅专栏

torch.save()用法：保存模型参数

filepath = os.path.join(args.save_path, 'checkpoint_model_epoch_{}.pth.tar'.format(epoch)) #最终参数模型
torch.save(state, filepath)

参考链接：
https://www.cnblogs.com/qinduanyinghua/p/9311410.html

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。