Tensorflow:保存,加載模型

有時常常需要將訓練的模型保存下來,用以預測或之後接着訓練。 1. 保存模型 主要是通過紅色框圈出的兩句話來保存模型的,下面是對這兩句話的解釋: saver = tf.train.Saver() 在定義saver類時,可以指定需要保存的變量,若不指定,則是保存計算圖中所有的變量,即像我這樣處理。max_to_keeo參數是指最多保存模型的個數,此處設置爲5,即最多保存5個模型。 saver.save
相關文章
相關標籤/搜索