pytorch-image-models/timm/scheduler
Ross Wightman 29a37e23ee LR scheduler update:
* add polynomial decay 'poly'
* cleanup cycle specific args for cosine, poly, and tanh sched, t_mul -> cycle_mul, decay -> cycle_decay, default cycle_limit to 1 in each opt
* add k-decay for cosine and poly sched as per https://arxiv.org/abs/2004.05909
* change default tanh ub/lb to push inflection to later epochs
2021-09-01 17:33:11 -07:00
..
__init__.py LR scheduler update: 2021-09-01 17:33:11 -07:00
cosine_lr.py LR scheduler update: 2021-09-01 17:33:11 -07:00
multistep_lr.py
plateau_lr.py
poly_lr.py LR scheduler update: 2021-09-01 17:33:11 -07:00
scheduler.py
scheduler_factory.py LR scheduler update: 2021-09-01 17:33:11 -07:00
step_lr.py
tanh_lr.py LR scheduler update: 2021-09-01 17:33:11 -07:00