Fix issue #6

Typo, `self.var` should be `self.running_var`
pulp-platform · Apr 23, 2020 · 0b5186f · 0b5186f
1 parent 1486960
commit 0b5186f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/quant/pact.py b/nemo/quant/pact.py
@@ -364,7 +364,7 @@ def reset_alpha(self, use_max=True, nb_std=5.):
         if use_max:
             self.alpha.data[0] = self.max.item()
         else:
-            self.alpha.data[0] = nb_std * torch.sqrt(self.var).item()
+            self.alpha.data[0] = nb_std * torch.sqrt(self.running_var).item()
 
     def get_statistics(self):
         r"""Returns the statistics collected up to now.