# Copyright (c) OpenMMLab. All rights reserved.
from .averaged_model import (ExponentialMovingAverage, MomentumAnnealingEMA,
                             StochasticWeightAverage)
from .wrappers import (MMDataParallel, MMDistributedDataParallel,
                       is_model_wrapper)

__all__ = [
    'MMDistributedDataParallel', 'MMDataParallel', 'is_model_wrapper',
    'StochasticWeightAverage', 'ExponentialMovingAverage',
    'MomentumAnnealingEMA'
]

# Copyright (c) OpenMMLab. All rights reserved.
from .wrappers import (MMDataParallel, MMDistributedDataParallel,
                       is_model_wrapper)

__all__ = ['MMDistributedDataParallel', 'MMDataParallel', 'is_model_wrapper']
