1. 程式人生 > >tf實現在上次訓練結果上繼續訓練

tf實現在上次訓練結果上繼續訓練

程式碼很簡單,只需要執行下面,自會自動將引數以及計算圖恢復,準備計算。

ckpt = tf.train.get_checkpoint_state(FLAGS.logs_dir)
if ckpt and ckpt.model_checkpoint_path:
saver.restore(sess, ckpt.model_checkpoint_path)#自動恢復model_checkpoint_path儲存模型一般是最新
print("Model restored...")