mmpretrain/mmcls/utils/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .collect_env import collect_env
from .device import auto_select_device
from .distribution import wrap_distributed_model, wrap_non_distributed_model
from .logger import get_root_logger, load_json_log
from .setup_env import setup_multi_processes

__all__ = [
    'collect_env', 'get_root_logger', 'load_json_log', 'setup_multi_processes',
    'wrap_non_distributed_model', 'wrap_distributed_model',
    'auto_select_device'
]