如何获取最新检查点图的文件名?
在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*)
?
答案 0 :(得分:0)
我刚刚找到了一篇文章,其中有人实施了我在问题中提到的例程。这不是一个“答案”,但是如果没有内置解决方案,我将使用它。
这摘自JoãoFelipe Santos在seaandsalor发表的精彩文章。
我不知道使用规则,所以我不想直接链接到他的代码。如果有兴趣,请点击该链接,然后转到底部提到的要点。