|
|
|
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) |
|
|
