激昂老虎

文章
3
资源
0
加入时间
3年0月21天

基于pytorch的保存和加载模型参数的方法

更多python教程请到: 菜鸟教程 https://www.piaodoo.com/ 当我们花费大量的精力训练完网络,下次预测数据时不想再(有时也不必再)训练一次时,这时候torch.save(),torch.load()就要登场了。保存和加载模型参数有两种方式:方式一:torch.save(net.state_dict(),path):功能:保存训练完的网络的各层参数(即weights和bias)其中:net.state_dict()获取各层参数,path是文件存放路径(通常保存文件格