what's in loss again

2018-11-04 11:45:44 +01:00
parent 0b7a20c651
commit 5ee171689c
1 changed files with 1 additions and 1 deletions
--- 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()