From 9f0ba7960eeb36f14685639fefd7daaa57a9183d Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Thu, 4 Nov 2021 16:18:26 +0800 Subject: [PATCH 1/2] add reference --- ppcls/utils/misc.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ppcls/utils/misc.py b/ppcls/utils/misc.py index 90dc47e9b..08ab7b6f7 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): From fa087f393619b8cb733e380e6eb48cd4d17a29a9 Mon Sep 17 00:00:00 2001 From: littletomatodonkey Date: Thu, 4 Nov 2021 16:29:12 +0800 Subject: [PATCH 2/2] fix ema reference --- ppcls/utils/ema.py | 5 +++++ 1 file changed, 5 insertions(+) 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