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)

你可能感兴趣的:(tensorflow,python,人工智能)