ott.neural.networks.velocity_field.ema.init_ema# ott.neural.networks.velocity_field.ema.init_ema(model)[source]# Create initial exponential moving average (EMA) state. Parameters: model (Module) – Model to average. Return type: Module Returns: Copy of the model with parameters set to 0s.
