Spaces:
Paused
Paused
# Copyright (c) OpenMMLab. All rights reserved. | |
from torch.nn.parallel import DataParallel, DistributedDataParallel | |
from annotator.mmpkg.mmcv.utils import Registry | |
MODULE_WRAPPERS = Registry('module wrapper') | |
MODULE_WRAPPERS.register_module(module=DataParallel) | |
MODULE_WRAPPERS.register_module(module=DistributedDataParallel) | |