Get a dfs module list of the given module. Its order is same as the order of creations of modules.
(module: nn.Module, memo: Optional[Set[nn.Module]] = None, prefix: str = "")
| 30 | |
| 31 | |
| 32 | def _get_dfs_module_list(module: nn.Module, memo: Optional[Set[nn.Module]] = None, prefix: str = ""): |
| 33 | """Get a dfs module list of the given module. Its order is same as the order of creations of modules.""" |
| 34 | if memo is None: |
| 35 | memo = set() |
| 36 | if module not in memo: |
| 37 | for name, submodule in module._modules.items(): |
| 38 | if submodule is None: |
| 39 | continue |
| 40 | submodule_prefix = prefix + ("." if prefix else "") + name |
| 41 | for m in _get_dfs_module_list(submodule, memo, submodule_prefix): |
| 42 | yield m |
| 43 | |
| 44 | memo.add(module) |
| 45 | yield prefix, module |
| 46 | |
| 47 | |
| 48 | def _get_shallow_copy_model(model: nn.Module): |
no test coverage detected
searching dependent graphs…