Spaces:
Runtime error
Runtime error
| # Copyright (c) OpenMMLab. All rights reserved. | |
| from .ascend_util import (batch_images_to_levels, | |
| get_max_num_gt_division_factor, masked_fill) | |
| from .collect_env import collect_env | |
| from .compat_config import compat_cfg | |
| from .logger import get_caller_name, get_root_logger, log_img_scale | |
| from .memory import AvoidCUDAOOM, AvoidOOM | |
| from .misc import find_latest_checkpoint, update_data_root | |
| from .replace_cfg_vals import replace_cfg_vals | |
| from .rfnext import rfnext_init_model | |
| from .setup_env import setup_multi_processes | |
| from .split_batch import split_batch | |
| from .util_distribution import build_ddp, build_dp, get_device | |
| __all__ = [ | |
| 'get_root_logger', 'collect_env', 'find_latest_checkpoint', | |
| 'update_data_root', 'setup_multi_processes', 'get_caller_name', | |
| 'log_img_scale', 'compat_cfg', 'split_batch', 'build_ddp', 'build_dp', | |
| 'get_device', 'replace_cfg_vals', 'AvoidOOM', 'AvoidCUDAOOM', | |
| 'get_max_num_gt_division_factor', 'masked_fill', 'batch_images_to_levels', | |
| 'rfnext_init_model' | |
| ] | |