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

mim*_*mic 2 tensorflow

我正在尝试使用带有检查点的预训练模型,但无法弄清楚应该将哪个文件作为检查点放入config。有文件:

  • 模型.ckpt.meta
  • model.ckpt.index
  • model.ckpt.data0000-of-0001

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

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

Jun*_*Lee 6

  • 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〜)(相对路径还是绝对路径)。