mirror of https://github.com/hpcaitech/ColossalAI
14 lines
632 B
Python
14 lines
632 B
Python
from .cosine import CosineAnnealingLR, CosineAnnealingWarmupLR, FlatAnnealingLR, FlatAnnealingWarmupLR
|
|
from .linear import LinearWarmupLR
|
|
from .multistep import MultiStepLR, MultiStepWarmupLR
|
|
from .onecycle import OneCycleLR
|
|
from .poly import PolynomialLR, PolynomialWarmupLR
|
|
from .torch import LambdaLR, MultiplicativeLR, StepLR, ExponentialLR
|
|
|
|
__all__ = [
|
|
'CosineAnnealingLR', 'CosineAnnealingWarmupLR', 'FlatAnnealingLR', 'FlatAnnealingWarmupLR', 'LinearWarmupLR',
|
|
'MultiStepLR', 'MultiStepWarmupLR', 'OneCycleLR', 'PolynomialLR', 'PolynomialWarmupLR', 'LambdaLR',
|
|
'MultiplicativeLR', 'StepLR',
|
|
'ExponentialLR'
|
|
]
|