github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[NFC] polish colossalai/nn/lr_scheduler/__init__.py (#1255) 

code style
pull/1298/head
Thunderbeee 2022-07-12 16:36:51 +08:00 committed by Frank Lee
parent 50f2ad213f
commit 9738fb0f78
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/nn/lr_scheduler/__init__.py
View File

@ -8,6 +8,5 @@ from .torch import LambdaLR, MultiplicativeLR, StepLR, ExponentialLR
__all__ = [
'CosineAnnealingLR', 'CosineAnnealingWarmupLR', 'FlatAnnealingLR', 'FlatAnnealingWarmupLR', 'LinearWarmupLR',
'MultiStepLR', 'MultiStepWarmupLR', 'OneCycleLR', 'PolynomialLR', 'PolynomialWarmupLR', 'LambdaLR',
'MultiplicativeLR', 'StepLR',
'ExponentialLR'
'MultiplicativeLR', 'StepLR', 'ExponentialLR'
]