Adopting exponential moving average (EMA) for PL pipeline

[YazanGhafir]

Hi, why not at the end of training_step?