Spaces:
Running on Zero
Running on Zero
File size: 364 Bytes
f0395ef | 1 2 3 4 5 6 7 8 9 10 | from .distributed import MMDistributedDataParallel
from .ddp_wrapper import DistributedDataParallelWrapper
from .fsdp_wrapper import FSDPWrapper
from .fsdp2_wrapper import FSDP2Wrapper
from .utils import apply_module_wrapper
__all__ = [
'MMDistributedDataParallel', 'DistributedDataParallelWrapper', 'FSDPWrapper', 'FSDP2Wrapper',
'apply_module_wrapper']
|