diff --git a/run_squad.py b/run_squad.py index 53334a68d..40905b3fe 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()