Enable V2 behavior in transformer_main.py.

The transformer model requires TF2. PiperOrigin-RevId: 266026991

Enable V2 behavior in transformer_main.py.
The transformer model requires TF2. PiperOrigin-RevId: 266026991
68c3c655 · Reed Wanderman-Milne · A. Unique TensorFlower · 503b1e61 · 68c3c655
Commit 68c3c655 authored Aug 28, 2019 by Reed Wanderman-Milne Committed by A. Unique TensorFlower Aug 28, 2019
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

official/transformer/v2/transformer_main.py official/transformer/v2/transformer_main.py +1 -0

No files found.
--- a/official/transformer/v2/transformer_main.py
+++ b/official/transformer/v2/transformer_main.py
@@ -464,6 +464,7 @@ def main(_):
 if __name__ == "__main__":
+  tf.compat.v1.enable_v2_behavior()
  logging.set_verbosity(logging.INFO)
  misc.define_transformer_flags()
  app.run(main)