检查点文件的加载方式/位置

Question

我正在运行tensorflow object_detection模型的model_main.py。 我想知道哪个.py文件具有代码，该模型在哪里检查“ mode_dir”中的现有检查点，以及该模型如何知道从保存的最新检查点开始加载/启动。

我签入了model_lib.py，但仍无法弄清。

Answer 1

除了上面的讨论之外，here是在tensorflow对象检测api中加载预训练的检查点的地方。如您所见，检查点路径由变量train_config.fine_tune_checkpoint传递，该变量是配置文件中train_config下的一个字段。它与model_dir不同，后者仅用于保存最新的检查点。

如果您使用同一命令行来恢复训练，则训练似乎会自动从model_dir中的最新检查点开始，并且tensorflow Supervisor支持此操作。请参阅here以供参考。