ott.neural.networks.velocity_field.unet.UNet.iter_modules

UNet.iter_modules()

Warning: this method is method is deprecated; use iter_modules() instead.

Recursively iterates over all nested Module’s of the current Module, including the current Module. Alias of iter_modules().

Return type:

Iterator[tuple[tuple[Key, ...], Module]]