Use tf.compat.v1.app
This commit is contained in:
parent
831989ccfb
commit
d55a60e534
|
@ -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()
|
||||
|
|
|
@ -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()
|
||||
|
|
|
@ -1215,4 +1215,4 @@ def main(_):
|
|||
|
||||
if __name__ == "__main__":
|
||||
FLAGS = extract_run_squad_flags()
|
||||
tf.app.run()
|
||||
tf.compat.v1.app.run()
|
Loading…
Reference in a new issue