|
|
|
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' |
|
] |
|
|
