mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix k_decay default arg != 1.0 in poly scheduler

Browse Source
This commit is contained in:
Ross Wightman 2021-11-22 09:37:36 -08:00
parent 947e1df3ef
commit f0507f6da6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/scheduler/poly_lr.py
View File

@ -37,7 +37,7 @@ class PolyLRScheduler(Scheduler):
noise_pct=0.67, noise_pct=0.67,
noise_std=1.0, noise_std=1.0,
noise_seed=42, noise_seed=42,
k_decay=.5, k_decay=1.0,
initialize=True) -> None: initialize=True) -> None:
super().__init__( super().__init__(
optimizer, param_group_field="lr", optimizer, param_group_field="lr",