Skip to content

GitLab

Unverified Commit 4b6fc6b8 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Fix bug on EMA n_averaged estimation. (#4544)

parent 8cc6d521
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
references/classification/utils.py
View file @ 4b6fc6b8
...@@ -180,7 +180,7 @@ class ExponentialMovingAverage(torch.optim.swa_utils.AveragedModel): ...@@ -180,7 +180,7 @@ class ExponentialMovingAverage(torch.optim.swa_utils.AveragedModel):
p_swa.detach().copy_(p_model_) p_swa.detach().copy_(p_model_)
else: else:
p_swa.detach().copy_(self.avg_fn(p_swa.detach(), p_model_, self.n_averaged.to(device))) p_swa.detach().copy_(self.avg_fn(p_swa.detach(), p_model_, self.n_averaged.to(device)))
self.n_averaged += 1 self.n_averaged += 1
def accuracy(output, target, topk=(1,)): def accuracy(output, target, topk=(1,)):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment