saver = tf.train.Saver(max_to_keep=None) 保存全部已訓練模型

通常用到saver保存模型,一般不加參數。 saver = tf.train.Saver() 昨天看到已經訓好的模型出現在列表中,過幾小時就消失了,感到很驚訝。 後來在網上一搜,發現 saver = tf.train.Saver()的默認參數爲: max_to_keep表示保存最近的幾個模型,設置爲None或者0 就是保存全部的模型。 keep_checkpoint_every_n_hours,每
相關文章
相關標籤/搜索