From f0507f6da6c3eb93960e1d712e8a40b81fd8d8fb Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 22 Nov 2021 09:37:36 -0800 Subject: [PATCH] Fix k_decay default arg != 1.0 in poly scheduler --- timm/scheduler/poly_lr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/scheduler/poly_lr.py b/timm/scheduler/poly_lr.py index 0c1e63b7..9c351be6 100644 --- a/timm/scheduler/poly_lr.py +++ b/timm/scheduler/poly_lr.py @@ -37,7 +37,7 @@ class PolyLRScheduler(Scheduler): noise_pct=0.67, noise_std=1.0, noise_seed=42, - k_decay=.5, + k_decay=1.0, initialize=True) -> None: super().__init__( optimizer, param_group_field="lr",