tensorboard 恢复预训练模型,在此基础上继续训练（对于.ckpt,.h）!

Posted 2021-05-16 星代码119

一、保存.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)