From 0ed9f9ddc9b93bbcd6d3828803f20a23cc500f5e Mon Sep 17 00:00:00 2001
From: Zehui Lin <zehui-lin_t@outlook.com>
Date: Thu, 21 Jan 2021 09:55:41 +0800
Subject: fix comments

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

(limited to 'torch_ema')

diff --git a/torch_ema/ema.py b/torch_ema/ema.py
index b819a04..8f39bc5 100644
--- a/torch_ema/ema.py
+++ b/torch_ema/ema.py
@@ -66,7 +66,7 @@ class ExponentialMovingAverage:
 
         Args: 
           parameters: Iterable of `torch.nn.Parameter`; the parameters to be
-            updated with the stored moving averages.
+            temporary stored in.
         """
         self.collected_parameters = []
         for param in parameters:
@@ -81,7 +81,7 @@ class ExponentialMovingAverage:
 
         Args: 
           parameters: Iterable of `torch.nn.Parameter`; the parameters to be
-            updated with the stored moving averages.
+            updated with the stored parameters.
         """
         for c_param, param in zip(self.collected_parameters, parameters):
             if param.requires_grad:
-- 
cgit v1.2.3