A simple way to keep track of an Exponential Moving Average (EMA) version of your pytorch model - View it on GitHub