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的,不会报错。