From ea26dae5de6be94126fc1744294872b3859d8ac4 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 11 Nov 2016 14:45:57 -0800 Subject: Fixed checkpoint restore conditions to work with both V1 and V2 checkpoints. Change: 138921188 --- tensorflow/models/rnn/translate/translate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/models/rnn/translate/translate.py b/tensorflow/models/rnn/translate/translate.py index ea96dc4796..9b6773917e 100644 --- a/tensorflow/models/rnn/translate/translate.py +++ b/tensorflow/models/rnn/translate/translate.py @@ -131,7 +131,7 @@ def create_model(session, forward_only): forward_only=forward_only, dtype=dtype) ckpt = tf.train.get_checkpoint_state(FLAGS.train_dir) - if ckpt and tf.gfile.Exists(ckpt.model_checkpoint_path): + if ckpt and tf.train.checkpoint_exists(ckpt.model_checkpoint_path): print("Reading model parameters from %s" % ckpt.model_checkpoint_path) model.saver.restore(session, ckpt.model_checkpoint_path) else: -- cgit v1.2.3