summaryrefslogtreecommitdiffstats
path: root/training/lr.py
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-01-01 19:19:52 +0100
committerVolpeon <git@volpeon.ink>2023-01-01 19:19:52 +0100
commitadc52fb8821a496bc8d78235bf10466b39df03e0 (patch)
tree8a6337a6ac10cbe76c55514ab559c647e69fb1aa /training/lr.py
parentFixed accuracy calc, other improvements (diff)
downloadtextual-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.py6
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,