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

PyTorch入门实战教程

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

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

方法一(推荐):

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

保存

恢复

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

方法二:

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

保存

恢复

一个相对完整的例子

saving

loading

获取模型中某些层的参数

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

Out:

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

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

PyTorch入门实战教程
除特别注明外,本站所有文章均为 PyTorch 中文网原创,转载请注明出处:https://www.pytorchtutorial.com/pytorch-note5-save-and-restore-models/

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部