I am confused as to how to apply EMA. Is the shadow model or the actual model used to process training items and compute gradients? If its the shadow model, then I don’t see how it’s different from using a smaller learning rate with the optimizer. Is it the case that the shadow model is only used at test time?