1. 程式人生 > >TensorFlow基礎學習——使用tf.train.Saver類儲存模型資訊

TensorFlow基礎學習——使用tf.train.Saver類儲存模型資訊

轉載請註明出處:http://blog.csdn.net/dongdong9223/article/details/83655798
本文出自【我是幹勾魚的部落格

Ingredient:

在TensorFlow中,tf.train.Saver類用來儲存模型。值得注意的是,在TensorFlow中的變數是儲存在於Session環境中的,也就是說,只有在Session環境下才會存有變數值,因此,儲存模型時需要傳入session,例如:

saver = tf.train.Saver(
) with tf.Session() as sess: tf.initialize_all_variables().run() saver.save(sess,"./checkpoint_dir/MyModel")

注意!TensorFlow預設只儲存最近的5個模型檔案,如果希望儲存更多,可以通過max_to_keep來指定。

例如,希望每2小時儲存1次模型,而且只是儲存最近的6個模型檔案。:

tf.train.Saver(max_to_keep=6, keep_checkpoint_every_n_hours=2)

參考:

Tensorflow載入預訓練模型和儲存模型

一份快速完整的Tensorflow模型儲存和恢復教程(譯)