diff options
author | Asim Shankar <ashankar@google.com> | 2018-07-19 08:56:07 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-19 09:02:54 -0700 |
commit | 2509b3a2152c8dda9fff8ed58f414c1316fa5379 (patch) | |
tree | 4660b2dc70d623d8270cff9545edacdcd534a8fd /third_party/examples | |
parent | e9e48b963b1ad1274ad8a0ad7d07d7fa990fe6b9 (diff) |
eager guide: s/tfe.Checkpoint/tf.train.Checkpoint/
PiperOrigin-RevId: 205248470
Diffstat (limited to 'third_party/examples')
-rw-r--r-- | third_party/examples/eager/spinn/spinn.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/third_party/examples/eager/spinn/spinn.py b/third_party/examples/eager/spinn/spinn.py index c242ef3fdd..de63ebe9e6 100644 --- a/third_party/examples/eager/spinn/spinn.py +++ b/third_party/examples/eager/spinn/spinn.py @@ -626,7 +626,7 @@ def train_or_infer_spinn(embed, model = SNLIClassifier(config, embed) global_step = tf.train.get_or_create_global_step() trainer = SNLIClassifierTrainer(model, config.lr) - checkpoint = tfe.Checkpoint(trainer=trainer, global_step=global_step) + checkpoint = tf.train.Checkpoint(trainer=trainer, global_step=global_step) checkpoint.restore(tf.train.latest_checkpoint(config.logdir)) if inference_sentence_pair: |