From e1422a34a3be89d02a0b943188190f41873f6538 Mon Sep 17 00:00:00 2001
From: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>
Date: Tue, 21 Jun 2022 14:40:00 +0800
Subject: [PATCH] [Fix]: Fix missing schedulers in __init__.py of schedulers
 (#319)

---
 mmengine/optim/__init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/mmengine/optim/__init__.py b/mmengine/optim/__init__.py
index c32a49be..9b34441e 100644
--- a/mmengine/optim/__init__.py
+++ b/mmengine/optim/__init__.py
@@ -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'
 ]