ott.neural.networks.velocity_field.ema.update_ema

Contents

ott.neural.networks.velocity_field.ema.update_ema#

ott.neural.networks.velocity_field.ema.update_ema(model, *, ema, decay)[source]#

Update the EMA of a model.

Parameters:

  • model (Module) – Model to average.

  • ema (Module) – EMA of the model.

  • decay (float) – Decay factor.

Return type:

None

Returns:

Nothing, just updates the EMA in-place.

Contents