what's in loss again

5ee17168 · thomwolf · 0b7a20c6 · 5ee17168
Commit 5ee17168 authored Nov 04, 2018 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

run_squad.py run_squad.py +1 -1

No files found.
--- 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()