From 8e9d62225db11913bf7ef67221fc3508d7fe1149 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Tue, 17 Jan 2023 16:39:33 +0100
Subject: Update

---
 training/lr.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'training/lr.py')

diff --git a/training/lr.py b/training/lr.py
index 902c4eb..9690738 100644
--- a/training/lr.py
+++ b/training/lr.py
@@ -101,7 +101,7 @@ class LRFinder():
 
                         self.accelerator.backward(loss)
 
-                        on_before_optimize(epoch)
+                        on_before_optimize(lr_scheduler.get_last_lr()[0], epoch)
 
                         self.optimizer.step()
                         lr_scheduler.step()
-- 
cgit v1.2.3-54-g00ecf