Skip to content

GitLab

Commit 68c3c655 authored by Reed Wanderman-Milne's avatar Reed Wanderman-Milne Committed by A. Unique TensorFlower
Browse files

Enable V2 behavior in transformer_main.py. 

The transformer model requires TF2.

PiperOrigin-RevId: 266026991
parent 503b1e61
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/transformer/v2/transformer_main.py
View file @ 68c3c655
...@@ -464,6 +464,7 @@ def main(_): ...@@ -464,6 +464,7 @@ def main(_):
if __name__ == "__main__": if __name__ == "__main__":
tf.compat.v1.enable_v2_behavior()
logging.set_verbosity(logging.INFO) logging.set_verbosity(logging.INFO)
misc.define_transformer_flags() misc.define_transformer_flags()
app.run(main) app.run(main)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment