diff options
Diffstat (limited to 'training/lr.py')
-rw-r--r-- | training/lr.py | 2 |
1 files changed, 1 insertions, 1 deletions
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(): | |||
101 | 101 | ||
102 | self.accelerator.backward(loss) | 102 | self.accelerator.backward(loss) |
103 | 103 | ||
104 | on_before_optimize(epoch) | 104 | on_before_optimize(lr_scheduler.get_last_lr()[0], epoch) |
105 | 105 | ||
106 | self.optimizer.step() | 106 | self.optimizer.step() |
107 | lr_scheduler.step() | 107 | lr_scheduler.step() |