From 5ee171689c07b05ef02ca6596f3ec7cfd247478e Mon Sep 17 00:00:00 2001 From: thomwolf Date: Sun, 4 Nov 2018 11:45:44 +0100 Subject: [PATCH] what's in loss again --- run_squad.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_squad.py b/run_squad.py index 53334a68d0..40905b3feb 100644 --- a/run_squad.py +++ b/run_squad.py @@ -873,7 +873,7 @@ def main(): logger.info("HHHHH Forward") loss, _ = model(input_ids, segment_ids, input_mask, start_positions, end_positions) model.zero_grad() - logger.info("HHHHH Backward") + logger.info("HHHHH Backward, loss: {}".format(loss)) loss.backward() logger.info("HHHHH Loading data") optimizer.step()