TensorFlow2 入门指南 | 19 模型文件的保存与加载

Posted 2022-02-25 AI 菌

前言：

本专栏在保证内容完整性的基础上，力求简洁，旨在让初学者能够更快地、高效地入门TensorFlow2 深度学习框架。如果觉得本专栏对您有帮助的话，可以给一个小小的三连，各位的支持将是我创作的最大动力！

系列文章汇总：TensorFlow2 入门指南
Github项目地址：https://github.com/Keyird/TensorFlow2-for-beginner

---

文章目录

• • 一、概要
  • 二、张量方式
  • 三、网络方式
  • 四、SavedModel 方式

---

一、概要

一般模型训练完成后，需要及时保存模型，方便后续进行模型测试和部署。模型的加载也就是在模型测试和部署过程中进行的。

实际上，在训练时间隔性地保存模型状态也是非常好的习惯，这一点对于训练大规模的网络尤其重要，一般大规模的网络需要训练数天乃至数周的时长