mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-10-22 07:14:09 +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']
|