diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2016-11-11 14:45:57 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-11-11 15:04:58 -0800 |
commit | ea26dae5de6be94126fc1744294872b3859d8ac4 (patch) | |
tree | 89592be0957cba66e6714236585fe7a88057621d | |
parent | bc516c8e9dcf6630bd0f162e605330b12ff854e4 (diff) |
Fixed checkpoint restore conditions to work with both V1 and V2 checkpoints.
Change: 138921188
-rw-r--r-- | tensorflow/models/rnn/translate/translate.py | 2 |
1 files changed, 1 insertions, 1 deletions
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: |