lucidrains/ema-pytorch · AI Repo StatsLUlucidrains/ema-pytorch
A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
Training & fine-tuningPython MIT 42created 2022-06-20 #artificial-intelligence#deep-learning#exponential-moving-average