children

UNet.children() → Iterator[torch.nn.modules.module.Module]

Returns an iterator over immediate children modules.

Yields

Module – a child module