|
|
|
from .registry import MODULE_WRAPPERS
|
|
|
|
|
|
def is_module_wrapper(module):
|
|
"""Check if a module is a module wrapper.
|
|
|
|
The following 3 modules in MMCV (and their subclasses) are regarded as
|
|
module wrappers: DataParallel, DistributedDataParallel,
|
|
MMDistributedDataParallel (the deprecated version). You may add you own
|
|
module wrapper by registering it to mmcv.parallel.MODULE_WRAPPERS.
|
|
|
|
Args:
|
|
module (nn.Module): The module to be checked.
|
|
|
|
Returns:
|
|
bool: True if the input module is a module wrapper.
|
|
"""
|
|
module_wrappers = tuple(MODULE_WRAPPERS.module_dict.values())
|
|
return isinstance(module, module_wrappers)
|
|
|
