tf.train.latest_checkpoint()是否从tf.train.import_meta_graph()继承模型名称

时间:2017-08-04 16:50:28

标签: machine-learning tensorflow

如果我有以下代码:

new_saver = tf.train.import_meta_graph('../logs/model-2.meta')
new_saver.restore(restored_sess,tf.train.latest_checkpoint('../logs/')) 

在我的../logs/文件中我有两个模型:model-1和model-2

TF会自动从model-2获取最新的检查点,因为它是在import_meta_graph中指定的。

将不同的模型保存在不同的文件夹中会更好吗?

1 个答案:

答案 0 :(得分:0)

您可以使用保护程序设置模型名称,以便为不同的模型saver.save(sess, 'model_name', global_step=epoch_number)使用单独的名称。对于文件夹,只保存一个checkpoint文件。这保留了tf.train.latest_checkpoint使用的有关最新模型的信息。

因此对于不同的模型,使用单独的文件夹总是好的。