diff options
| author | Volpeon <git@volpeon.ink> | 2023-06-21 13:28:49 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-06-21 13:28:49 +0200 |
| commit | 8364ce697ddf6117fdd4f7222832d546d63880de (patch) | |
| tree | 152c99815bbd8b2659d0dabe63c98f63151c97c2 /training/lr.py | |
| parent | Fix LoRA training with DAdan (diff) | |
| download | textual-inversion-diff-8364ce697ddf6117fdd4f7222832d546d63880de.tar.gz textual-inversion-diff-8364ce697ddf6117fdd4f7222832d546d63880de.tar.bz2 textual-inversion-diff-8364ce697ddf6117fdd4f7222832d546d63880de.zip | |
Update
Diffstat (limited to 'training/lr.py')
| -rw-r--r-- | training/lr.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/training/lr.py b/training/lr.py index f5b362f..a75078f 100644 --- a/training/lr.py +++ b/training/lr.py | |||
| @@ -23,12 +23,12 @@ def plot_metrics( | |||
| 23 | fig, ax_loss = plt.subplots() | 23 | fig, ax_loss = plt.subplots() |
| 24 | ax_acc = ax_loss.twinx() | 24 | ax_acc = ax_loss.twinx() |
| 25 | 25 | ||
| 26 | ax_loss.plot(lrs, losses, color='red') | 26 | ax_loss.plot(lrs, losses, color="red") |
| 27 | ax_loss.set_xscale("log") | 27 | ax_loss.set_xscale("log") |
| 28 | ax_loss.set_xlabel(f"Learning rate") | 28 | ax_loss.set_xlabel(f"Learning rate") |
| 29 | ax_loss.set_ylabel("Loss") | 29 | ax_loss.set_ylabel("Loss") |
| 30 | 30 | ||
| 31 | ax_acc.plot(lrs, accs, color='blue') | 31 | ax_acc.plot(lrs, accs, color="blue") |
| 32 | ax_acc.set_xscale("log") | 32 | ax_acc.set_xscale("log") |
| 33 | ax_acc.set_ylabel("Accuracy") | 33 | ax_acc.set_ylabel("Accuracy") |
| 34 | 34 | ||