From 359c155723f0b08f6e6c1784c7951e33d92b306d Mon Sep 17 00:00:00 2001
From: Zehui Lin <zehui-lin_t@outlook.com>
Date: Mon, 8 Feb 2021 10:09:36 +0800
Subject: minor

---
 torch_ema/ema.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'torch_ema')

diff --git a/torch_ema/ema.py b/torch_ema/ema.py
index 8f39bc5..50f9aea 100644
--- a/torch_ema/ema.py
+++ b/torch_ema/ema.py
@@ -20,6 +20,7 @@ class ExponentialMovingAverage:
         """
         if decay < 0.0 or decay > 1.0:
             raise ValueError('Decay must be between 0 and 1')
+        self.collected_parameters = []
         self.decay = decay
         self.num_updates = 0 if use_num_updates else None
         self.shadow_params = [p.clone().detach()
@@ -54,9 +55,7 @@ class ExponentialMovingAverage:
           parameters: Iterable of `torch.nn.Parameter`; the parameters to be
             updated with the stored moving averages.
         """
-        self.collected_parameters = []
         for s_param, param in zip(self.shadow_params, parameters):
-            self.collected_parameters.append(param.clone())
             if param.requires_grad:
                 param.data.copy_(s_param.data)
 
-- 
cgit v1.2.3