From d5e4198955960c7a99d2a5f666bbcee99f3a9371 Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Mon, 10 Oct 2022 12:47:20 +0200
Subject: TI: Use different lr scheduler

---
 textual_inversion.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'textual_inversion.py')

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():
     parser.add_argument(
         "--lr_scheduler",
         type=str,
-        default="linear",
+        default="cosine",
         help=(
             'The scheduler type to use. Choose between ["linear", "cosine", "cosine_with_restarts", "polynomial",'
             ' "constant", "constant_with_warmup"]'
-- 
cgit v1.2.3-54-g00ecf