ott.neural.networks.velocity_field.ema.EMA.iter_modules

EMA.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]]