diff --git a/nemo/quant/pact.py b/nemo/quant/pact.py index f586fa8..2ef2822 100644 --- 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.