Skip to content

Commit 95ee60d

Browse files
authored
Fix bug on EMA n_averaged estimation. (#4544) (#4545)
# Conflicts: # references/classification/utils.py
1 parent 1a969c6 commit 95ee60d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

references/classification/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def update_parameters(self, model):
181181
else:
182182
p_swa.detach().copy_(self.avg_fn(p_swa.detach(), p_model_,
183183
self.n_averaged.to(device)))
184-
self.n_averaged += 1
184+
self.n_averaged += 1
185185

186186

187187
def accuracy(output, target, topk=(1,)):

0 commit comments

Comments
 (0)