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():
     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"]'