Files
ColossalAI/colossalai/nn/parallel/__init__.py
2022-06-06 15:34:41 +08:00

4 lines
82 B
Python

from .data_parallel import ColoDDP, ColoDDPV2
__all__ = ['ColoDDP', 'ColoDDPV2']