Tensorflow - 这是真正的检查点文件?

时间:2017-11-20 00:37:32

标签: tensorflow

我正在尝试使用带有检查点的预训练模型,但我无法确定哪个文件应作为检查点放入配置中。有文件:

  • model.ckpt.meta
  • model.ckpt.index
  • model.ckpt.data0000-的-0001

我尝试了所有这些,但我看到了错误。在不同的文章中,我只看到“model.ckpt”,但没有这样的文件。

我尝试从这里使用ssd_mobilenet:https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md

1 个答案:

答案 0 :(得分:4)

  • model.ckpt.meta:存储图形信息的文件
  • model.ckpt.index:存储变量索引的文件
  • model.ckpt.data0000-of-0001:存储变量值的文件

这三个文件都是真正的检查点文件。使用tf.train.Saver.restore恢复模型时,参数" save_path"应该是" ~model.ckpt"。您的错误可能是由于save_path无效而发生的。检查save_path中的三个文件(.meta,.index,.data~)(相对路径或绝对路径)。