2017-08-04 62 views
0

在我有下面的代碼的情況下:是否tf.train.latest_checkpoint()繼承tf.train.import_meta_graph型號名稱()

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中指定。

在不同的文件夾中保存不同的模型會更好嗎?

回答

0

您可以使用保存程序設置型號名稱,以便爲不同的型號使用不同的名稱saver.save(sess, 'model_name', global_step=epoch_number)。對於一個文件夾,只有一個checkpoint文件被保存。這保留了tf.train.latest_checkpoint所使用的最新型號的信息。

因此,對於不同的型號,使用單獨的文件夾總是很好。