Return a layer (nn.Module Object) given a hierarchical layer name, separated by /. Args: layer_name (str): the name of the layer.
(self, layer_name)
| 176 | self._register_hooks() |
| 177 | |
| 178 | def _get_layer(self, layer_name): |
| 179 | """ |
| 180 | Return a layer (nn.Module Object) given a hierarchical layer name, separated by /. |
| 181 | Args: |
| 182 | layer_name (str): the name of the layer. |
| 183 | """ |
| 184 | layer_ls = layer_name.split("/") |
| 185 | prev_module = self.model |
| 186 | for layer in layer_ls: |
| 187 | prev_module = prev_module._modules[layer] |
| 188 | |
| 189 | return prev_module |
| 190 | |
| 191 | def _register_single_hook(self, layer_name): |
| 192 | """ |
nothing calls this directly
no outgoing calls
no test coverage detected