A simple way to keep track of an Exponential Moving Average (EMA) version of your pytorch model
Full Changelog: https://github.com/lucidrains/ema-pytorch/compare/0.4.4...0.4.5
Full Changelog: https://github.com/lucidrains/ema-pytorch/compare/0.4.3...0.4.4
Full Changelog: https://github.com/lucidrains/ema-pytorch/compare/0.4.1...0.4.2
Full Changelog: https://github.com/lucidrains/ema-pytorch/compare/0.4.0...0.4.1
Full Changelog: https://github.com/lucidrains/ema-pytorch/compare/0.3.3...0.4.0
Full Changelog: https://github.com/lucidrains/ema-pytorch/compare/0.3.2...0.3.3