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/optimization.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/optimization.py')
| -rw-r--r-- | training/optimization.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/optimization.py b/training/optimization.py index 3340544..a79944f 100644 --- a/training/optimization.py +++ b/training/optimization.py | |||
| @@ -15,7 +15,7 @@ def get_one_cycle_schedule( | |||
| 15 | warmup: Literal["cos", "linear"] = "cos", | 15 | warmup: Literal["cos", "linear"] = "cos", |
| 16 | annealing: Literal["cos", "half_cos", "linear"] = "cos", | 16 | annealing: Literal["cos", "half_cos", "linear"] = "cos", |
| 17 | warmup_exp: int = 1, | 17 | warmup_exp: int = 1, |
| 18 | annealing_exp: int = 2, | 18 | annealing_exp: int = 1, |
| 19 | min_lr: int = 0.04, | 19 | min_lr: int = 0.04, |
| 20 | mid_point: int = 0.3, | 20 | mid_point: int = 0.3, |
| 21 | last_epoch: int = -1 | 21 | last_epoch: int = -1 |