oneflow.nn.Module.named_children

Module.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)