github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-29 14:30:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
code style
This commit is contained in:
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'
]