MnistModel.named_children#
- MnistModel.named_children() Iterator[Tuple[str, 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)