From 4f1da87c0a0b51be79869138dcf1685bc260ae29 Mon Sep 17 00:00:00 2001
From: Shakti Kumar <shakti.shrivastava13@gmail.com>
Date: Mon, 30 Mar 2020 14:41:19 -0400
Subject: [PATCH] adding current lr to stats for monitoring

Signed-off-by: Shakti Kumar <shakti.shrivastava13@gmail.com>
---
 train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 825f957..991e4dd 100644
--- 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.