[src] Fix cumulative normalization (asteroid-team#649)

iput2021-someya · Nov 11, 2022 · 9c543ec · 9c543ec
1 parent 140733b
commit 9c543ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/asteroid/masknn/norms.py b/asteroid/masknn/norms.py
@@ -98,7 +98,7 @@ def forward(self, x, EPS: float = 1e-8):
             start=chan, end=chan * (spec_len + 1), step=chan, dtype=x.dtype, device=x.device
         ).view(1, 1, -1)
         cum_mean = cum_sum / cnt
-        cum_var = cum_pow_sum - cum_mean.pow(2)
+        cum_var = cum_pow_sum / cnt - cum_mean.pow(2)
         return self.apply_gain_and_bias((x - cum_mean) / (cum_var + EPS).sqrt())