将张量流检查点恢复到以前的状态

时间:2018-08-08 00:41:32

标签: python tensorflow

我有一个包含以下内容的检查点目录:

checkpoint
mymodel.model-19002.data-00000-of-00001
mymodel.model-19002.index
mymodel.model-19002.meta
mymodel.model-20002.data-00000-of-00001
mymodel.model-20002.index
mymodel.model-20002.meta
mymodel.model-21002.data-00000-of-00001
mymodel.model-21002.index
mymodel.model-21002.meta

我正在使用它来恢复我的模型:

checkpoint_dir = "checkpointBlue"
with tf.Session(config=tfconfig) as sess:
   ckpt = tf.train.get_checkpoint_state(checkpoint_dir)


   print("restoring ckpt_name {}".format(ckpt_name))
   tf.train.Saver().restore(sess, os.path.join(checkpoint_dir, ckpt_name))

是否可以加载19002数据而不是最新的21002数据?

有时候我不小心训练了我的模型,我想回滚。

0 个答案:

没有答案