TensorFlow基礎學習——使用tf.train.Saver類儲存模型資訊
阿新 • • 發佈:2018-11-27
轉載請註明出處:http://blog.csdn.net/dongdong9223/article/details/83655798
本文出自【我是幹勾魚的部落格】
Ingredient:
- Python:Python 3.6.6(Python Downloads)
在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)