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