您的位置 首页 PyTorch 教程

[莫烦 PyTorch 系列教程] 3.4 – 保存和恢复模型

PyTorch入门实战教程

训练好了一个模型, 我们当然想要保存它, 留到下次要用的时候直接提取直接用, 这就是这节的内容啦. 我们用回归的神经网络举例实现保存提取.

保存

我们快速地建造数据, 搭建网络:

接下来我们有两种途径来保存

提取网络

这种方式将会提取整个神经网络, 网络大的时候可能会比较慢.

只提取网络参数

这种方式将会提取所有的参数, 然后再放到你的新建网络中.

显示结果

调用上面建立的几个功能, 然后出图.

这样我们就能看出三个网络完全一模一样啦.

所以这也就是在我 github 代码 中的每一步的意义啦.

文章来源:莫烦

PyTorch入门实战教程

发表评论

电子邮件地址不会被公开。 必填项已用*标注

评论列表(1)

返回顶部