Use tf.compat.v1.app

This commit is contained in:
Chang Lan 2021-05-05 00:17:48 -07:00
parent 831989ccfb
commit d55a60e534
3 changed files with 3 additions and 3 deletions

View file

@ -416,4 +416,4 @@ if __name__ == "__main__":
flags.mark_flag_as_required("bert_config_file")
flags.mark_flag_as_required("init_checkpoint")
flags.mark_flag_as_required("output_file")
tf.app.run()
tf.compat.v1.app.run()

View file

@ -311,4 +311,4 @@ if __name__ == "__main__":
flags.mark_flag_as_required("task_name")
flags.mark_flag_as_required("bert_hub_module_handle")
flags.mark_flag_as_required("output_dir")
tf.app.run()
tf.compat.v1.app.run()

View file

@ -1215,4 +1215,4 @@ def main(_):
if __name__ == "__main__":
FLAGS = extract_run_squad_flags()
tf.app.run()
tf.compat.v1.app.run()