fix shouldn't normalize

kymatio · Oct 24, 2024 · 758378a · 758378a
1 parent 9da8628
commit 758378a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/murenn/dtcwt/nn.py b/murenn/dtcwt/nn.py
@@ -185,6 +185,5 @@ def __init__(self, J_phi):
     def forward(self, x):
         for j in range(self.J_phi):
             x, _ = self.phi(x)
-            # Normalize the coefficients
-            x = x[:,:,::2] / math.sqrt(2)
+            x = x[:,:,::2]
         return x