Tensorflow加载预训练模型和保存模型

Posted 2022-02-07 laokanblog

1. Tensorflow模型文件

（1）checkpoint
该文件是文本文件，里面记录了保存的最新的checkpoint文件以及其他checkpoint文件列表。在测试的时候，可以通过修改这个文件，指定具体使用哪个模型
（2）meta文件
这个文件保存的是计算图结构，可以理解为神经网络结构图。是一个二进制的pb格式文件，包含变量、op、集合等。
（3）ckpt文件
ckpt文件是二进制文件，保存了所有的weights、biases、gradients等变量。在tensorflow 0.11之前，保存在.ckpt文件中。0.11后，通过两个文件保存

• 一个是data文件：保存了所有的变量
• 一个是index文件：保存了所有变量（包括权重、偏置、梯度等的值）是一个key-value的字典。

2. 保存Tensorflow模型

参考