Skip to content

GitLab

Commit 70e14f03 authored by Zongwei Zhou's avatar Zongwei Zhou Committed by A. Unique TensorFlower
Browse files

Enable run_eagerly for BERT SQUAD 

PiperOrigin-RevId: 292604336
parent 4d6820e0
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/nlp/bert/run_squad.py
View file @ 70e14f03
...@@ -416,7 +416,7 @@ def main(_): ...@@ -416,7 +416,7 @@ def main(_):
all_reduce_alg=FLAGS.all_reduce_alg, all_reduce_alg=FLAGS.all_reduce_alg,
tpu_address=FLAGS.tpu) tpu_address=FLAGS.tpu)
if FLAGS.mode in ('train', 'train_and_predict'): if FLAGS.mode in ('train', 'train_and_predict'):
train_squad(strategy, input_meta_data) train_squad(strategy, input_meta_data, run_eagerly=FLAGS.run_eagerly)
if FLAGS.mode in ('predict', 'train_and_predict'): if FLAGS.mode in ('predict', 'train_and_predict'):
predict_squad(strategy, input_meta_data) predict_squad(strategy, input_meta_data)
......
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