(layer, name, new_module)
| 102 | return {name: m for name, m in module.named_modules() if isinstance(m, type)} |
| 103 | |
| 104 | def set_op_by_name(layer, name, new_module): |
| 105 | levels = name.split('.') |
| 106 | if len(levels) > 1: |
| 107 | mod_ = layer |
| 108 | for l_idx in range(len(levels)-1): |
| 109 | if levels[l_idx].isdigit(): |
| 110 | mod_ = mod_[int(levels[l_idx])] |
| 111 | else: |
| 112 | mod_ = getattr(mod_, levels[l_idx]) |
| 113 | setattr(mod_, levels[-1], new_module) |
| 114 | else: |
| 115 | setattr(layer, name, new_module) |
| 116 | |
| 117 | # def add_new_module(name, original_module, added_module): |
| 118 | # levels = name.split('.') |
no outgoing calls
no test coverage detected