adding current lr to stats for monitoring

Signed-off-by: Shakti Kumar <[email protected]>
jerrodparker20 · Mar 30, 2020 · 4f1da87 · 4f1da87
1 parent 3fcbe83
commit 4f1da87
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -905,12 +905,13 @@ def checkpoint():
             #     best_val_loss = val_loss
 
             logging.info(
-                "Steps %i @ %.1f SPS. Loss %f. %sStats:\n%s",
+                "Steps %i @ %.1f SPS. Loss %f. %sStats:\n%s\n%s",
                 step,
                 sps,
                 total_loss,
                 mean_return,
                 pprint.pformat(stats),
+                optimizer.params_group[0]['lr']
             )
     except KeyboardInterrupt:
         return  # Try joining actors then quit.