保存提取 - PyTorch | 莫烦Python
切换视频源:

保存提取

作者: 莫烦 编辑: 莫烦 2017-05-06

学习资料:

要点

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

保存

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

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

提取网络

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

只提取网络参数

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

显示结果

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

3-4-1.png

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

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


降低知识传递的门槛

莫烦很常从互联网上学习知识,开源分享的人是我学习的榜样。 他们的行为也改变了我对教育的态度: 降低知识传递的门槛免费 奉献我的所学正是受这种态度的影响。 通过 【赞助莫烦】 能让我感到认同,我也更有理由坚持下去。