diff options
| author | Zehui Lin <zehui-lin_t@outlook.com> | 2021-01-21 09:55:41 +0800 |
|---|---|---|
| committer | Zehui Lin <zehui-lin_t@outlook.com> | 2021-01-21 09:55:41 +0800 |
| commit | 0ed9f9ddc9b93bbcd6d3828803f20a23cc500f5e (patch) | |
| tree | ecccc4cadf90bb824fd52e57393e5a92a9ad4034 /torch_ema/ema.py | |
| parent | 205c76e4789709094ab7bb658b02f9abb789a66a (diff) | |
fix comments
Diffstat (limited to 'torch_ema/ema.py')
| -rw-r--r-- | torch_ema/ema.py | 4 |
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: |