[hotfix] add hybrid adam to __init__ (#584)

[hotfix] add hybrid adam to init (#584)
104cbbb3 · ver217 · GitHub · e6d50ec1 · 104cbbb3
Unverified Commit 104cbbb3 authored Mar 31, 2022 by ver217 Committed by GitHub Mar 31, 2022
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

colossalai/nn/optimizer/__init__.py colossalai/nn/optimizer/__init__.py +2 -3

No files found.
--- a/colossalai/nn/optimizer/__init__.py
+++ b/colossalai/nn/optimizer/__init__.py
@@ -5,7 +5,6 @@ from .fused_sgd import FusedSGD
 from .lamb import Lamb
 from .lars import Lars
 from .cpu_adam import CPUAdam
+from .hybrid_adam import HybridAdam
-__all__ = [
+__all__ = ['ColossalaiOptimizer', 'FusedLAMB', 'FusedAdam', 'FusedSGD', 'Lamb', 'Lars', 'CPUAdam', 'HybridAdam']
-    'ColossalaiOptimizer', 'FusedLAMB', 'FusedAdam', 'FusedSGD', 'Lamb', 'Lars', 'CPUAdam'
-]