removed steps per loop parameter from squad.

Now Training loss showed per step
kamalkraj · Dec 5, 2019 · 5a620f3 · 5a620f3
1 parent fa90194
commit 5a620f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_squad.py b/run_squad.py
@@ -736,7 +736,7 @@ def train_squad(strategy,
         model_dir=FLAGS.model_dir,
         train_input_fn=train_input_fn,
         steps_per_epoch=steps_per_epoch,
-        steps_per_loop=steps_per_epoch,
+        # steps_per_loop=steps_per_epoch,
         epochs=FLAGS.num_train_epochs,
         run_eagerly=run_eagerly,
         custom_callbacks=custom_callbacks)