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的,不會報錯。