【问题标题】:Tensorflow - which is the real checkpoint file?Tensorflow - 哪个是真正的检查点文件?
【发布时间】:2017-11-20 11:47:16
【问题描述】:

我正在尝试使用带有检查点的预训练模型,但我不知道应该将哪个文件作为检查点放入配置中。有文件:

  • model.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

【问题讨论】:

  • 还是试试“model.ckpt”吧。这不识别文件,但对于 TensorFlow,它识别检查点路径。
  • 谢谢,但没用。我有一堆错误,看起来很像“..MobilenetV1/Conv2d_6_depthwise/BatchNorm/gamma/RMSProp_1 not found in checkpoint”
  • 好的。如果您提供用于访问检查点的代码,您可能会得到更多有用的答案。
  • @MattScarpino 你是对的,我只需要指向不存在的文件,包括包含所有这些东西的文件夹。这对我来说没有意义(因为这个文件尚未创建但它可以工作)。

标签: tensorflow


【解决方案1】:
  • 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~)。

【讨论】:

  • 是的,你是对的。我实际上放错了路径。即使没有这样的文件,它也应该指向文件夹/model.ckpt。这实际上对我来说没有任何意义,因为即使在我最终开始培训之后,这个文件也从未创建过。
猜你喜欢
  • 1970-01-01
  • 2016-11-10
  • 2021-12-04
  • 1970-01-01
  • 2020-05-03
  • 2019-04-03
  • 2017-07-30
  • 1970-01-01
相关资源
最近更新 更多