flamo 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

flamo/auxiliary/reverb.py

@@ -410,7 +410,7 @@ class parallelGFDNAccurateGEQ(parallelFDNAccurateGEQ):
             device=device
         )
         self.n_gains = self.size[0]
-        self.size = (self.n_groups * self.size[0], len(self.delays))
+        self.size = (self.n_groups * self.size[0],)
         self.param = torch.nn.Parameter(
             torch.empty(self.size, device=self.device), requires_grad=self.requires_grad
         )
@@ -420,13 +420,13 @@ class parallelGFDNAccurateGEQ(parallelFDNAccurateGEQ):
         r"""
         Computes the polynomial coefficients for the SOS section.
         """
-        a = torch.zeros((3, self.size[0]+1, len(self.delays)), device=self.device)
-        b = torch.zeros((3, self.size[0]+1, len(self.delays)), device=self.device)
+        a = torch.zeros((3, self.size[0]+self.n_groups, len(self.delays)), device=self.device)
+        b = torch.zeros((3, self.size[0]+self.n_groups, len(self.delays)), device=self.device)
         for n_i in range(len(self.delays)):
             for i_group in range(self.n_groups):
                 (
-                    b[:, i_group * (self.n_gains) : (i_group + 1) * self.n_gains, n_i],
-                    a[:, i_group * (self.n_gains) : (i_group + 1) * self.n_gains, n_i],
+                    b[:, i_group * (self.n_gains+1) : (i_group + 1) * (self.n_gains+1), n_i],
+                    a[:, i_group * (self.n_gains+1) : (i_group + 1) * (self.n_gains+1), n_i],
                 ) = accurate_geq(
                     target_gain=param[
                         i_group * self.n_gains : (i_group + 1) * self.n_gains, n_i

flamo/optimize/surface.py

@@ -484,9 +484,9 @@ class LossSurface(LossProfile):
                 )
             for i_crit in range(len(self.criteria)):
                 if i_plot == 0:
-                    x = loss[..., i_crit].mean(0)
+                    x = loss[..., i_crit].mean(0).T
                 else:
-                    x = loss[..., i_crit].std(0)
+                    x = loss[..., i_crit].std(0).T
                 criterion_name = self.criteria[i_crit].name
                 if len(self.criteria) == 1:
 

{flamo-0.1.10.dist-info → flamo-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flamo
-Version: 0.1.10
+Version: 0.1.12
 Summary: An Open-Source Library for Frequency-Domain Differentiable Audio Processing
 Project-URL: Homepage, https://github.com/gdalsanto/flamo
 Project-URL: Issues, https://github.com/gdalsanto/flamo/issues

{flamo-0.1.10.dist-info → flamo-0.1.12.dist-info}/RECORD

@@ -5,20 +5,20 @@ flamo/auxiliary/__init__.py,sha256=7lVNh8OxHloZ4KPmp-iTUJnUbi8XbuRzGaQ3Z-NKXio,4
 flamo/auxiliary/eq.py,sha256=eIWMIq0ggizXLhTdeWWbgBXWUFXCJyoEbkBH7Gzasao,6779
 flamo/auxiliary/filterbank.py,sha256=02w8dI8HoNDtKpdVhSJkIkd-h-KNXvZtivf3l4_ozzU,9866
 flamo/auxiliary/minimize.py,sha256=fMTAAAk9yD7Y4luKS4XA1-HTq44xo2opq_dRPRrhlIY,2474
-flamo/auxiliary/reverb.py,sha256=dcfTlUnLZ42ZdguNwm2GFYijh5-Yje52I3yH7VobkOw,34750
+flamo/auxiliary/reverb.py,sha256=fAJgTGq_yaQguBrrnifJNl4ySIkPb0NdrcFKNCNL4xc,34769
 flamo/auxiliary/scattering.py,sha256=qlK8cynrpde56yLlbPuScC0Y1VmsPb0SFXl6Xisv6hA,9420
 flamo/auxiliary/velvet.py,sha256=B4pYEnhaQPkh02pxqiGdAhLRX2g-eWtHezphi0_h4Qs,4201
 flamo/auxiliary/config/config.py,sha256=CxXj-8sLq0_m9KyLg1a6NwLoK1UvTz3i0jZOLraq14I,2893
 flamo/optimize/__init__.py,sha256=grgxLmQ7m-c9MvRdIejmEAaaajfBwgeaZAv2qjHIvPw,65
 flamo/optimize/dataset.py,sha256=2mfzsnyX_bzavXouII9ee_pd6ti4lv215ieGJHscceI,5829
 flamo/optimize/loss.py,sha256=h6EeqjdX5P1SqDBKBavSxV25VBgnYK8tuX91wk6lw_g,33466
-flamo/optimize/surface.py,sha256=uvsgxLFSvJ18s8kPcb22G3W1rgycXP1nNX0q48Pda2g,26135
+flamo/optimize/surface.py,sha256=eGDg9Zksicu1EO3irjyECEhp2oplj9HMNNuDK_WnHe4,26139
 flamo/optimize/trainer.py,sha256=he4nUjLC-3RTlxxBIw33r5k8mQfgAGvN1wpPBAWCjVo,12045
 flamo/optimize/utils.py,sha256=R5-KoZagRho3eykY88pC3UB2mc5SsE4Yv9X-ogskXdA,1610
 flamo/processor/__init__.py,sha256=paGdxGVZgA2VAs0tBwRd0bobzGxeyK79DS7ZGO8drkI,41
 flamo/processor/dsp.py,sha256=n92YJPrES-ydwHgXmZ9RkFevIC3n-Wh4X8I1QNZqcV0,126378
 flamo/processor/system.py,sha256=9XwLtaGEVs9glVOFvyiPnQpsnR_Wjrv6k1i1qCs8D1Q,42516
-flamo-0.1.10.dist-info/METADATA,sha256=zaSuPqxqlbFNOUVtZiyMhaqd2WBuBeGfW6UXn-3PiaY,7826
-flamo-0.1.10.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flamo-0.1.10.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
-flamo-0.1.10.dist-info/RECORD,,
+flamo-0.1.12.dist-info/METADATA,sha256=03qRhouKUzsmZs4bXb3LUGyhVAUIYLiZk6Z7KOiYyyk,7826
+flamo-0.1.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flamo-0.1.12.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
+flamo-0.1.12.dist-info/RECORD,,