Skip to content
Snippets Groups Projects
Unverified Commit e1422a34 authored by Jiazhen Wang's avatar Jiazhen Wang Committed by GitHub
Browse files

[Fix]: Fix missing schedulers in __init__.py of schedulers (#319)

parent e470c3aa
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
mmengine/optim/__init__.py
+ 4
− 2
View file @ e1422a34
......@@ -10,7 +10,8 @@ from .scheduler import (ConstantLR, ConstantMomentum, ConstantParamScheduler,
LinearLR, LinearMomentum, LinearParamScheduler,
MultiStepLR, MultiStepMomentum,
MultiStepParamScheduler, OneCycleLR,
OneCycleParamScheduler, StepLR, StepMomentum,
OneCycleParamScheduler, PolyLR, PolyMomentum,
PolyParamScheduler, StepLR, StepMomentum,
StepParamScheduler, _ParamScheduler)
# yapf: enable
......@@ -23,5 +24,6 @@ __all__ = [
'CosineAnnealingParamScheduler', 'ExponentialParamScheduler',
'LinearParamScheduler', 'MultiStepParamScheduler', 'StepParamScheduler',
'_ParamScheduler', 'OptimWrapper', 'AmpOptimWrapper', 'OptimWrapperDict',
'OneCycleParamScheduler', 'OneCycleLR'
'OneCycleParamScheduler', 'OneCycleLR', 'PolyLR', 'PolyMomentum',
'PolyParamScheduler'
]
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

Datenschutz und Impressum