diff --git a/ppcls/utils/ema.py b/ppcls/utils/ema.py
index e41f472d6..b54cdb1b2 100644
--- a/ppcls/utils/ema.py
+++ b/ppcls/utils/ema.py
@@ -17,6 +17,11 @@ import numpy as np
 
 
 class ExponentialMovingAverage():
+    """
+    Exponential Moving Average
+    Code was heavily based on https://github.com/Wanger-SJTU/SegToolbox.Pytorch/blob/master/lib/utils/ema.py
+    """
+
     def __init__(self, model, decay, thres_steps=True):
         self._model = model
         self._decay = decay