diff options
Diffstat (limited to 'third_party/examples/eager/spinn/spinn.py')
| -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 8a2b24aa4e..67456a5bdf 100644 --- a/third_party/examples/eager/spinn/spinn.py +++ b/third_party/examples/eager/spinn/spinn.py @@ -462,7 +462,7 @@ class SNLIClassifierTrainer(tfe.Checkpointable): 2. logits as a dense `Tensor` of shape (batch_size, d_out), where d_out is the output dimension size of the SNLIClassifier. """ - with tfe.GradientTape() as tape: + with tf.GradientTape() as tape: tape.watch(self._model.variables) logits = self._model(premise, premise_transition, |