TensorFlow 中的 .ckpt文件

tf.train.Saver()

  • 使用 tf.train.Saver() 保存参数通常会生成以下文件

在这里插入图片描述

各文件作用

  • .ckpt文件:是旧版本的输出saver.save(sess),相当于你的.ckpt-data
  • “checkpoint”:文件仅用于告知某些TF函数,这是最新的检查点文件。
  • .ckpt-meta:包含元图,即计算图的结构,没有变量的值(基本上你可以在tensorboard / graph中看到)。
  • .ckpt-data:包含所有变量的值,没有结构。
  • .ckpt-index:可能是内部需要的某种索引来正确映射前两个文件,它通常不是必需的
  • 你可以只用 .ckpt-meta 和 .ckpt-data 恢复一个模型

python中恢复模型

  • 要在python中恢复模型,通常会使用元数据和数据文件(但也可以使用该.pb文件):
saver = tf.train.import_meta_graph(path_to_ckpt_meta) saver.restore(sess, path_to_ckpt_data)

该.pb文件可以保存您的整个图表(元+数据)

要在c ++中加载和使用(但不训练)图形,您通常会使用它来创建freeze_graph,它会.pb从元数据和数据创建文件。
要小心,(至少在以前的TF版本和某些人中)py提供的功能freeze_graph不能正常工作,所以你必须使用脚本版本。

Reference

https://blog.csdn.net/nanhuaibeian/article/details/101751439

猜你喜欢

转载自blog.csdn.net/m0_38024592/article/details/111477723
今日推荐