tensorflow1.6 繼續訓練操作

sess=tf.Session()

sess.run(tf.global_variables_initializer)

ckpt= tf.train.get_checkpoint_state(check_point_path)#這裏的check_point_path是checkpoint所在的文件夾路徑

saver=tf.train.Saver()

if ckpt and ckpt.model_checkpoint_path:

    saver.restore(sess,ckpt.model_checkpoint_path)

train...

發佈了36 篇原創文章 · 獲贊 17 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章