/pytorch_ema

Tiny PyTorch library for maintaining a moving average of a collection of parameters.

Primary LanguagePythonMIT LicenseMIT

Stargazers

No one’s star this repository yet.