Add EMA to DecayAdagrad (#57866)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/57866
As titled
Test Plan: f271267365
Reviewed By: lanlanfb
Differential Revision: D28292875
fbshipit-source-id: f6532048eb558afce87fdada3b7dfa8457a1f538