diff options
| author | Volpeon <git@volpeon.ink> | 2022-10-10 12:47:20 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2022-10-10 12:47:20 +0200 |
| commit | d5e4198955960c7a99d2a5f666bbcee99f3a9371 (patch) | |
| tree | f59b963446574604eb5290e5437473def6997bae | |
| parent | Dreambooth: Add EMA support (diff) | |
| download | textual-inversion-diff-d5e4198955960c7a99d2a5f666bbcee99f3a9371.tar.gz textual-inversion-diff-d5e4198955960c7a99d2a5f666bbcee99f3a9371.tar.bz2 textual-inversion-diff-d5e4198955960c7a99d2a5f666bbcee99f3a9371.zip | |
TI: Use different lr scheduler
| -rw-r--r-- | textual_inversion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/textual_inversion.py b/textual_inversion.py index 7373472..b01bdbc 100644 --- a/textual_inversion.py +++ b/textual_inversion.py | |||
| @@ -140,7 +140,7 @@ def parse_args(): | |||
| 140 | parser.add_argument( | 140 | parser.add_argument( |
| 141 | "--lr_scheduler", | 141 | "--lr_scheduler", |
| 142 | type=str, | 142 | type=str, |
| 143 | default="linear", | 143 | default="cosine", |
| 144 | help=( | 144 | help=( |
| 145 | 'The scheduler type to use. Choose between ["linear", "cosine", "cosine_with_restarts", "polynomial",' | 145 | 'The scheduler type to use. Choose between ["linear", "cosine", "cosine_with_restarts", "polynomial",' |
| 146 | ' "constant", "constant_with_warmup"]' | 146 | ' "constant", "constant_with_warmup"]' |