1 files changed, 6 insertions, 0 deletions

diff --git a/training/lr.py b/training/lr.py
index 0c5ce9e..3abd2f2 100644
--- a/training/lr.py
+++ b/training/lr.py
@@ -102,6 +102,12 @@ class LRFinder():
             losses.append(loss)
             accs.append(acc)
 
+            self.accelerator.log({
+                "loss": loss,
+                "acc": acc,
+                "lr": lr,
+            }, step=epoch)
+
             progress_bar.set_postfix({
                 "loss": loss,
                 "loss/best": best_loss,