Enable run_eagerly for BERT SQUAD

PiperOrigin-RevId: 292604336

Enable run_eagerly for BERT SQUAD
PiperOrigin-RevId: 292604336
70e14f03 · Zongwei Zhou · A. Unique TensorFlower · 4d6820e0 · 70e14f03
Commit 70e14f03 authored Jan 31, 2020 by Zongwei Zhou Committed by A. Unique TensorFlower Jan 31, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/nlp/bert/run_squad.py official/nlp/bert/run_squad.py +1 -1

No files found.
--- a/official/nlp/bert/run_squad.py
+++ b/official/nlp/bert/run_squad.py
@@ -416,7 +416,7 @@ def main(_):
      all_reduce_alg=FLAGS.all_reduce_alg,
      tpu_address=FLAGS.tpu)
  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'):
    predict_squad(strategy, input_meta_data)