starling.models.ema.EMA

class EMA[source]

Bases: object

Methods

`__init__`
`reset_parameters`
`step_ema`
`update_average`
`update_model_average`

__init__(beta: float = 0.999)[source]

update_model_average(ma_model, current_model)[source]

update_average(ema_model_params, current_model_params)[source]

step_ema(ema_model, model, step_start_ema=2000)[source]

reset_parameters(ema_model, model)[source]