fadel / pytorch_ema

Tiny PyTorch library for maintaining a moving average of a collection of parameters.
MIT License
400 stars 25 forks source link