mmdetection/mmdet/utils/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .collect_env import collect_env
from .compat_config import compat_cfg
from .dist_utils import (all_reduce_dict, allreduce_grads, reduce_mean,
                         sync_random_seed)
from .logger import get_caller_name, log_img_scale
from .memory import AvoidCUDAOOM, AvoidOOM
from .misc import (find_latest_checkpoint, get_test_pipeline_cfg,
                   update_data_root)
from .replace_cfg_vals import replace_cfg_vals
from .setup_env import (register_all_modules, setup_cache_size_limit_of_dynamo,
                        setup_multi_processes)
from .split_batch import split_batch
from .typing_utils import (ConfigType, InstanceList, MultiConfig,
                           OptConfigType, OptInstanceList, OptMultiConfig,
                           OptPixelList, PixelList, RangeType)

__all__ = [
    'collect_env', 'find_latest_checkpoint', 'update_data_root',
    'setup_multi_processes', 'get_caller_name', 'log_img_scale', 'compat_cfg',
    'split_batch', 'register_all_modules', 'replace_cfg_vals', 'AvoidOOM',
    'AvoidCUDAOOM', 'all_reduce_dict', 'allreduce_grads', 'reduce_mean',
    'sync_random_seed', 'ConfigType', 'InstanceList', 'MultiConfig',
    'OptConfigType', 'OptInstanceList', 'OptMultiConfig', 'OptPixelList',
    'PixelList', 'RangeType', 'get_test_pipeline_cfg',
    'setup_cache_size_limit_of_dynamo'
]