diff --git a/ppcls/utils/misc.py b/ppcls/utils/misc.py
index 90dc47e9bb7e6162ad921a14d24bc5467a415b62..08ab7b6f77cb85b0a822713ee7d573d561762d14 100644
--- a/ppcls/utils/misc.py
+++ b/ppcls/utils/misc.py
@@ -18,6 +18,7 @@ __all__ = ['AverageMeter']
 class AverageMeter(object):
     """
     Computes and stores the average and current value
+    Code was based on https://github.com/pytorch/examples/blob/master/imagenet/main.py
     """
 
     def __init__(self, name='', fmt='f', postfix="", need_avg=True):