mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-05-02 08:37:36 +00:00
5 lines
147 B
Python
5 lines
147 B
Python
from .sharded_optim import ShardedOptimizer
|
|
from .sharded_optim_v2 import ShardedOptimizerV2
|
|
|
|
__all__ = ['ShardedOptimizer', 'ShardedOptimizerV2']
|