我有一个包含以下内容的检查点目录:
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数据?
有时候我不小心训练了我的模型,我想回滚。