diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-17 16:39:33 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-17 16:39:33 +0100 |
| commit | 8e9d62225db11913bf7ef67221fc3508d7fe1149 (patch) | |
| tree | 4c17e8491a77bc92deb276dedba7949a8bb7297a /training/lr.py | |
| parent | Optimized embedding normalization (diff) | |
| download | textual-inversion-diff-8e9d62225db11913bf7ef67221fc3508d7fe1149.tar.gz textual-inversion-diff-8e9d62225db11913bf7ef67221fc3508d7fe1149.tar.bz2 textual-inversion-diff-8e9d62225db11913bf7ef67221fc3508d7fe1149.zip | |
Update
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() |