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.