/ema-pytorch

A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model

Primary LanguagePythonMIT LicenseMIT

Issues