Build module from config or return the module itself. Args: module (Union[dict, nn.Module]): The module to build. builder (Registry): The registry to build module. *args, **kwargs: Arguments passed to build function. Returns: Any: The built module.
(module, builder, **kwargs)
| 5 | |
| 6 | |
| 7 | def build_module(module, builder, **kwargs): |
| 8 | """Build module from config or return the module itself. |
| 9 | |
| 10 | Args: |
| 11 | module (Union[dict, nn.Module]): The module to build. |
| 12 | builder (Registry): The registry to build module. |
| 13 | *args, **kwargs: Arguments passed to build function. |
| 14 | |
| 15 | Returns: |
| 16 | Any: The built module. |
| 17 | """ |
| 18 | if isinstance(module, dict): |
| 19 | cfg = deepcopy(module) |
| 20 | for k, v in kwargs.items(): |
| 21 | cfg[k] = v |
| 22 | return builder.build(cfg) |
| 23 | elif isinstance(module, nn.Module): |
| 24 | return module |
| 25 | elif module is None: |
| 26 | return None |
| 27 | else: |
| 28 | raise TypeError(f"Only support dict and nn.Module, but got {type(module)}.") |
| 29 | |
| 30 | |
| 31 | MODELS = Registry( |