From 5a620f39d5ef5950651429122011034641c046a2 Mon Sep 17 00:00:00 2001 From: kamalkraj Date: Thu, 5 Dec 2019 05:01:04 -0800 Subject: [PATCH] removed steps per loop parameter from squad. Now Training loss showed per step --- run_squad.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_squad.py b/run_squad.py index 8b74fb4..7389979 100644 --- 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)