From adc52fb8821a496bc8d78235bf10466b39df03e0 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Sun, 1 Jan 2023 19:19:52 +0100
Subject: Updates

---
 training/lr.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'training/lr.py')

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,
-- 
cgit v1.2.3-54-g00ecf