pytorch模型存储的2种实现方法
更新时间:2020年02月14日 12:00:56 作者:慢行厚积
今天小编就为大家分享一篇pytorch模型存储的2种实现方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
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'))
以上这篇pytorch模型存储的2种实现方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
相关文章
python chinesecalendar报错:"no available data for ye
这篇文章主要介绍了python chinesecalendar报错:“no available data for year {}, only year between [{}, {}] supported“的相关知识,需要的朋友可以参考下2023-03-03python之plt.hist函数的输入参数和返回值的用法解释
这篇文章主要介绍了python之plt.hist函数的输入参数和返回值的用法解释,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-10-10
最新评论