TensorFlow入門(九)使用 tf.train.Saver()保存模型

關於模型保存的一點心得 saver = tf.train.Saver(max_to_keep=3) 在定義 saver 的時候通常會定義最多保存模型的數量,通常來講,若是模型自己很大,咱們須要考慮到硬盤大小。若是你須要在當前訓練好的模型的基礎上進行 fine-tune,那麼儘量多的保存模型,後繼 fine-tune 不必定從最好的 ckpt 進行,由於有可能一會兒就過擬合了。可是若是保存太多,硬盤
相關文章
相關標籤/搜索