aboutsummaryrefslogtreecommitdiff
path: root/torch_ema/ema.py
diff options
context:
space:
mode:
authorZehui Lin <zehui-lin_t@outlook.com>2021-01-21 09:55:41 +0800
committerZehui Lin <zehui-lin_t@outlook.com>2021-01-21 09:55:41 +0800
commit0ed9f9ddc9b93bbcd6d3828803f20a23cc500f5e (patch)
treeecccc4cadf90bb824fd52e57393e5a92a9ad4034 /torch_ema/ema.py
parent205c76e4789709094ab7bb658b02f9abb789a66a (diff)
fix comments
Diffstat (limited to 'torch_ema/ema.py')
-rw-r--r--torch_ema/ema.py4
1 files changed, 2 insertions, 2 deletions
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: