PyTorch 学习笔记(五):存储和恢复模型并查看参数

在模型完成训练后,我们需要将训练好的模型保存为一个文件供测试使用,或者因为一些原因我们需要继续之前的状态训练之前保存的模型,那么如何在PyTorch中保存和恢复模型呢?

参考PyTorch官方的这份repo,我们知道有两种方法可以实现我们想要的效果。

方法一(推荐):

第一种方法也是官方推荐的方法,只保存和恢复模型中的参数。

保存

恢复

使用这种方法,我们需要自己导入模型的结构信息。

方法二:

使用这种方法,将会保存模型的参数和结构信息。

保存

恢复

一个相对完整的例子

saving

loading

获取模型中某些层的参数

对于恢复的模型,如果我们想查看某些层的参数,可以:

Out:

如果我们想获取conv1的weight和bias:

文章来源:http://www.aiboy.pub/2017/06/05/How_To_Save_And_Restore_Model/

PyTorch入门实战教程
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论