/pytorch_ema

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

Primary LanguagePythonMIT LicenseMIT