add reference

2021-11-04 16:18:26 +08:00 · 2021-11-04 16:18:26 +08:00 · cae027605c
parent d1d818f5a2
commit cae027605c
1 changed files with 1 additions and 0 deletions
--- 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):