Tensorflow: Saver / tf.train.latest_checkpoint 与相对路径不能很好地工作

创建于 2015-12-20  ·  3评论  ·  资料来源: tensorflow/tensorflow

例如,如果通过执行saver.save(session, '/path/to/test')将模型保存到文件夹test - 将导致具有checkpointsmodel-0文件。 checkpoints文件将包含如下内容:

model_checkpoint_path: "test/model-0"
all_model_checkpoint_paths: "test/model-0"

如果您现在从同一目录调用tf.train.latest_checkpoint('test') ,将导致None

这是由于os.path.join(checkpoint_dir, ckpt.model_checkpoint_path)对于像上面这样的相对路径将导致test/test/model-0

最有用的评论

当您更改运行文件夹的名称时,此问题仍然存在。 似乎检查点/检查点文件只保留旧路径,并且当在这个新文件夹上使用 latest_checkpoint 时,tf 失败。

所有3条评论

修正者:
更改 ID:I47d8536b9b2ed3dcc193d6e6b7f4573a4e22c9b3

我们放弃了 gerrit 一个——如果https://github.com/tensorflow/tensorflow/pull/573看起来不错,我们可以合并

当您更改运行文件夹的名称时,此问题仍然存在。 似乎检查点/检查点文件只保留旧路径,并且当在这个新文件夹上使用 latest_checkpoint 时,tf 失败。

此页面是否有帮助?
0 / 5 - 0 等级