一、背景
tensorflow还是非常灵活的,但是有门槛的,门槛有两点api比较多,理解tensorflow的思维(先建立图,然后建立session,然后运算,pytorch是不用这样),这篇文章说说保存tensorflow保存的模型
Tensorflow的保存分为四种:
- 1. checkpoint模式;
- 2. saved_model模式(包含pb文件和variables);
- 3. 纯pb模式;(只有一个pb文件)
- 4.keras的 h5 模式
1. checkpoint模式
# 一个完整的模型保存包含下面四个文件
checkpoint: 具有最近检查点列表的协议缓存区
.data: 保存模型中的变量
.index: 标识检查点
.meta: 保存模型中计算图的结构信息