TensorFlow-深度學習-12-訓練模型保存與使用

有這麼一種情況,如果說我訓練了一個數據集,效果還不錯,但總不能每次都要經過一次訓練在拿來用吧,一次訓練能達到上萬次,對於普通電腦來說根本不可能訓練,所以我們要把它保存成一個模型文件(檢查點),用到的時候直接拿來用就好了。 保存模型文件很簡單,只需要用下面2句話就行了: saver = tf.train.Saver() saver.save(sess,save_path="checkpoin
相關文章
相關標籤/搜索