diff options
author | Volpeon <git@volpeon.ink> | 2023-01-01 19:19:52 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-01-01 19:19:52 +0100 |
commit | adc52fb8821a496bc8d78235bf10466b39df03e0 (patch) | |
tree | 8a6337a6ac10cbe76c55514ab559c647e69fb1aa /training/lr.py | |
parent | Fixed accuracy calc, other improvements (diff) | |
download | textual-inversion-diff-adc52fb8821a496bc8d78235bf10466b39df03e0.tar.gz textual-inversion-diff-adc52fb8821a496bc8d78235bf10466b39df03e0.tar.bz2 textual-inversion-diff-adc52fb8821a496bc8d78235bf10466b39df03e0.zip |
Updates
Diffstat (limited to 'training/lr.py')
-rw-r--r-- | training/lr.py | 6 |
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(): | |||
102 | losses.append(loss) | 102 | losses.append(loss) |
103 | accs.append(acc) | 103 | accs.append(acc) |
104 | 104 | ||
105 | self.accelerator.log({ | ||
106 | "loss": loss, | ||
107 | "acc": acc, | ||
108 | "lr": lr, | ||
109 | }, step=epoch) | ||
110 | |||
105 | progress_bar.set_postfix({ | 111 | progress_bar.set_postfix({ |
106 | "loss": loss, | 112 | "loss": loss, |
107 | "loss/best": best_loss, | 113 | "loss/best": best_loss, |