tensorflow學習:模型的保存與恢復(saver)

將訓練好的模型參數保存起來,以便以後進行驗證或測試,這是我們經常要做的事情。tf裏面提供模型保存的是tf.train.Saver()模塊。 模型保存,先要創建一個Saver對象:如 saver=tf.train.Saver() 在創建這個Saver對象的時候,有一個參數我們經常會用到,就是 max_to_keep 參數,這個是用來設置保存模型的個數,默認爲5,即 max_to_keep=5,保存最
相關文章
相關標籤/搜索