保存提取

学习资料:

• 本节的全部代码
• Tensorflow 的保存读取代码
• 我制作的 什么是神经网络 动画简介
• PyTorch 官网

要点¶

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

保存¶

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

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

提取网络¶

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

只提取网络参数¶

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

显示结果¶

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

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

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