Fix k_decay default arg != 1.0 in poly scheduler

pull/842/head
Ross Wightman 3 years ago
parent 947e1df3ef
commit f0507f6da6

@ -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",

Loading…
Cancel
Save