mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
* 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
9 lines
291 B
Python
9 lines
291 B
Python
from .cosine_lr import CosineLRScheduler
|
|
from .multistep_lr import MultiStepLRScheduler
|
|
from .plateau_lr import PlateauLRScheduler
|
|
from .poly_lr import PolyLRScheduler
|
|
from .step_lr import StepLRScheduler
|
|
from .tanh_lr import TanhLRScheduler
|
|
|
|
from .scheduler_factory import create_scheduler
|