tensorboard 恢复预训练模型,在此基础上继续训练（对于.ckpt,.h）!
Posted 星代码119
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了tensorboard 恢复预训练模型,在此基础上继续训练（对于.ckpt,.h）!相关的知识，希望对你有一定的参考价值。
一、保存.ckpt模型
保存模型，max_to_keep指的是保存最近的10个模型
self.saver = tf.train.Saver(tf.global_variables(), max_to_keep=10)
放在训练代码的最后，ckpt_file：需要保存ckpt的路径：
self.saver.save(self.sess, ckpt_file, global_step=epoch)
二，恢复.ckpt模型
如果全部恢复的话代码如下，self.initial_weigh为你的.ckpt文件路径，我这里是：
"./checkpoint/yolov3_coco_demo.ckpt"
恢复代码路径：
self.loader = tf.train.Saver()
self.loader.restore(self.sess, self.initial_weight)
以上是关于tensorboard 恢复预训练模型,在此基础上继续训练（对于.ckpt,.h）!的主要内容，如果未能解决你的问题，请参考以下文章
Keras 中的 TensorBoard 回调不尊重适合的初始时期？