TensorFlow检查点目录中图.meta的文件名

时间:2018-06-27 17:09:16

标签: tensorflow

如何获取最新检查点图的文件名?

tf.train.import_meta_graph的示例中,我通常会看到硬编码的文件名类似于checkpoint_dir +'model.ckpt-1234.meta'。

导入后,.restore可以加载最新的训练参数,如下所示:

saver.restore(*session*, tf.train.latest_checkpoint(*my_dir*))

但是,获取图形文件名的可靠方法是什么?就我而言,

tf.train_import_meta_data(tf.train.latest_checkpoint(*my_dir*) + '.meta')

应该做这项工作,但我认为它并不可靠,因为检查点不一定每次都保存元图,对吗?

我可以编写一个例程,该例程查看检查点目录并向后走,直到找到.meta。但是,是否有更好的/内置的方式,例如tf.train.latest_metagraph(*my_dir*)

1 个答案:

答案 0 :(得分:0)

我刚刚找到了一篇文章,其中有人实施了我在问题中提到的例程。这不是一个“答案”,但是如果没有内置解决方案,我将使用它。

这摘自JoãoFelipe Santos在seaandsalor发表的精彩文章。

我不知道使用规则,所以我不想直接链接到他的代码。如果有兴趣,请点击该链接,然后转到底部提到的要点。