fix bug

kymatio · Dec 6, 2024 · a0546c1 · a0546c1
1 parent e7c5cac
commit a0546c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/murenn/dtcwt/nn.py b/murenn/dtcwt/nn.py
@@ -99,7 +99,7 @@ def to_conv1d(self):
         device = self.conv1d[0].weight.data.device
         T = self.T  # Filter length
         J = self.dtcwt.J  # Number of levels of decomposition
-        N = 2 ** J * max(T, len(self.dtcwt.g0a))  # Hybrid filter length
+        N = 2 ** J * max(T, self.dtcwt.g0a.shape[-1]) * 2  # Hybrid filter length
 
         # Generate a zero signal
         x = torch.zeros(1, self.in_channels, N).to(device)