Fix k_decay default arg != 1.0 in poly scheduler

2021-11-22 09:37:36 -08:00 · 2021-11-22 09:37:36 -08:00 · f0507f6da6
parent 947e1df3ef
commit f0507f6da6
1 changed files with 1 additions and 1 deletions
--- 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",