mmclassification/mmcls/registry/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .registry import (BATCH_AUGMENTS, DATA_SAMPLERS, DATASETS, HOOKS, LOOPS,
                       METRICS, MODEL_WRAPPERS, MODELS,
                       OPTIM_WRAPPER_CONSTRUCTORS, OPTIM_WRAPPERS, OPTIMIZERS,
                       PARAM_SCHEDULERS, RUNNER_CONSTRUCTORS, RUNNERS,
                       TASK_UTILS, TRANSFORMS, VISBACKENDS, VISUALIZERS,
                       WEIGHT_INITIALIZERS)

__all__ = [
    'RUNNERS', 'RUNNER_CONSTRUCTORS', 'HOOKS', 'DATASETS', 'DATA_SAMPLERS',
    'TRANSFORMS', 'MODELS', 'WEIGHT_INITIALIZERS', 'OPTIMIZERS',
    'OPTIM_WRAPPERS', 'OPTIM_WRAPPER_CONSTRUCTORS', 'TASK_UTILS',
    'PARAM_SCHEDULERS', 'METRICS', 'MODEL_WRAPPERS', 'LOOPS', 'VISBACKENDS',
    'VISUALIZERS', 'BATCH_AUGMENTS'
]