From d5e4198955960c7a99d2a5f666bbcee99f3a9371 Mon Sep 17 00:00:00 2001 From: Volpeon 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(-) 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-70-g09d2