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, ...@@ -10,7 +10,8 @@ from .scheduler import (ConstantLR, ConstantMomentum, ConstantParamScheduler,
LinearLR, LinearMomentum, LinearParamScheduler, LinearLR, LinearMomentum, LinearParamScheduler,
MultiStepLR, MultiStepMomentum, MultiStepLR, MultiStepMomentum,
MultiStepParamScheduler, OneCycleLR, MultiStepParamScheduler, OneCycleLR,
OneCycleParamScheduler, StepLR, StepMomentum, OneCycleParamScheduler, PolyLR, PolyMomentum,
PolyParamScheduler, StepLR, StepMomentum,
StepParamScheduler, _ParamScheduler) StepParamScheduler, _ParamScheduler)
# yapf: enable # yapf: enable
...@@ -23,5 +24,6 @@ __all__ = [ ...@@ -23,5 +24,6 @@ __all__ = [
'CosineAnnealingParamScheduler', 'ExponentialParamScheduler', 'CosineAnnealingParamScheduler', 'ExponentialParamScheduler',
'LinearParamScheduler', 'MultiStepParamScheduler', 'StepParamScheduler', 'LinearParamScheduler', 'MultiStepParamScheduler', 'StepParamScheduler',
'_ParamScheduler', 'OptimWrapper', 'AmpOptimWrapper', 'OptimWrapperDict', '_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