004. tf.restore()函數的加載預訓練模型的記憶點

tensorflow裏,與pb模型不同的是,ckpt沒有固化模型的各張量參數,因此更利於在預訓練模型上繼續跑模型,優化模型。

恢復ckpt模型的代碼爲:

build_graph()
saver = tf.train.Saver()
saver.restore(sess, ckpt_state.model_checkpoint_path)

一個值得記憶的點:假設預訓練的ckpt模型是通過20個epoches,200個batches跑出來;在此基礎上新跑的代碼是可以設置不同的epoches和batches的,不會報錯。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章