tensorflow saver.restore()

加載pretrained_model模型

-需要注意的是,tensorflow提供了下列函數,可以將pretrained_model設爲文件夾,自動檢查並加載上次save的ckpt文件。
- 下面是facenet訓練時的加載上次模型的代碼。需要在源代碼中修改


saver = tf.train.Saver(tf.trainable_variables(), max_to_keep=3)
ckpt = tf.train.get_checkpoint_state(args.pretrained_model)
print('Restoring pretrained model: %s' % args.pretrained_model)
saver.restore(sess, ckpt.model_checkpoint_path)

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