composer.algorithms.ema#

Modules

 composer.algorithms.ema.ema Core Exponential Moving Average (EMA) classes and functions.

Exponential moving average maintains a moving average of model parameters and uses these at test time.

See the Method Card for more details.

Functions

 compute_ema Updates the weights of ema_model to be closer to the weights of model according to an exponential weighted average.

Classes

 EMA Maintains a shadow model with weights that follow the exponential moving average of the trained model weights.