__init__.py 648 Bytes
# Copyright (c) OpenMMLab. All rights reserved.
from .dist_utils import (DistOptimizerHook, all_reduce_dict, allreduce_grads,
                         reduce_mean, sync_random_seed)
from .misc import (center_of_mass, filter_scores_and_topk, flip_tensor,
                   generate_coordinate, mask2ndarray, multi_apply,
                   select_single_mlvl, unmap)

__all__ = [
    'allreduce_grads', 'DistOptimizerHook', 'reduce_mean', 'multi_apply',
    'unmap', 'mask2ndarray', 'flip_tensor', 'all_reduce_dict',
    'center_of_mass', 'generate_coordinate', 'select_single_mlvl',
    'filter_scores_and_topk', 'sync_random_seed'
]