summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2022-10-10 12:47:20 +0200
committerVolpeon <git@volpeon.ink>2022-10-10 12:47:20 +0200
commitd5e4198955960c7a99d2a5f666bbcee99f3a9371 (patch)
treef59b963446574604eb5290e5437473def6997bae
parentDreambooth: Add EMA support (diff)
downloadtextual-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.py2
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"]'