From a0546c1696e0a81e946e1074a8dfe1c89852d489 Mon Sep 17 00:00:00 2001
From: Xiran <xiranfr@gmail.com>
Date: Fri, 6 Dec 2024 14:39:15 +0100
Subject: [PATCH] fix bug

---
 murenn/dtcwt/nn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/murenn/dtcwt/nn.py b/murenn/dtcwt/nn.py
index 9ad03a0..bce8370 100644
--- 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)