From c24a24c77a825e71de2e67c1515c84d2b77701fa Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 2 Jan 2023 11:27:38 +0100 Subject: Improved one cycle scheduler --- train_ti.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'train_ti.py') diff --git a/train_ti.py b/train_ti.py index 870bd40..775b918 100644 --- a/train_ti.py +++ b/train_ti.py @@ -250,7 +250,7 @@ def parse_args(): parser.add_argument( "--lr_annealing_exp", type=int, - default=2, + default=1, help='If lr_annealing_func is "half_cos" or "cos", exponent to modify the function' ) parser.add_argument( -- cgit v1.2.3-54-g00ecf