named_children
- FullyConnected.named_children() Iterator[Tuple[str, torch.nn.modules.module.Module]]
Returns an iterator over immediate children modules, yielding both the name of the module as well as the module itself.
- Yields
(string, Module) – Tuple containing a name and child module
Example:
>>> for name, module in model.named_children(): >>> if name in ['conv4', 'conv5']: >>> print(module)