flamo 0.2.1__py3-none-any.whl → 0.2.3__py3-none-any.whl

flamo/optimize/surface.py

@@ -142,6 +142,7 @@ class LossProfile:
                 figsize=((len(self.criteria) + 2) // 3 * 5, 15),
             )
         steps = self.steps
+        steps = steps.cpu().numpy()
         for i_crit in range(len(self.criteria)):
             mean_loss = loss[:, :, i_crit].mean(0)
             std_loss = loss[:, :, i_crit].std(0)
@@ -308,6 +309,7 @@ class LossProfile:
         audio distances are bad at pitch,” arXiv preprint arXiv:2012.04572, 2020.
         """
         steps = self.steps
+        steps = steps.cpu().numpy()
         # find the index in steps of the element closest to the target value
         target_indx = np.abs(steps - self.param_config.target_value).argmin()
         accuracy = np.empty(loss.shape)
@@ -470,7 +472,7 @@ class LossSurface(LossProfile):
         steps_0 = self.steps_0
         steps_1 = self.steps_1
         # create a meshgrid
-        steps_0, steps_1 = np.meshgrid(steps_0, steps_1)
+        steps_0, steps_1 = np.meshgrid(steps_0.cpu().numpy(), steps_1.cpu().numpy())
         title = ["mean", "std"]
         for i_plot in range(2):
             if len(self.criteria) == 1:
@@ -634,6 +636,8 @@ class LossSurface(LossProfile):
         """
         steps_0 = self.steps_0
         steps_1 = self.steps_1
+        steps_0 = steps_0.cpu().numpy()
+        steps_1 = steps_1.cpu().numpy()
         # find the index in steps of the element closest to the target value
         target_indx_0 = np.abs(steps_0 - self.param_config[0].target_value).argmin()
         target_indx_1 = np.abs(steps_1 - self.param_config[1].target_value).argmin()

{flamo-0.2.1.dist-info → flamo-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flamo
-Version: 0.2.1
+Version: 0.2.3
 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.2.1.dist-info → flamo-0.2.3.dist-info}/RECORD

@@ -12,13 +12,13 @@ flamo/auxiliary/config/config.py,sha256=sZ3XvqwV6KiIc2n8HRtg7YJE3zhc7Vqblbqs-Z0b
 flamo/optimize/__init__.py,sha256=grgxLmQ7m-c9MvRdIejmEAaaajfBwgeaZAv2qjHIvPw,65
 flamo/optimize/dataset.py,sha256=6DviKGBuT8HZp6dm4ipOpd8XZeycIl1ci4QChx2iL0M,6556
 flamo/optimize/loss.py,sha256=h6EeqjdX5P1SqDBKBavSxV25VBgnYK8tuX91wk6lw_g,33466
-flamo/optimize/surface.py,sha256=aYWXg2ZNJzFjHRILcgrTfNkQ2VSoaMfMSnu9IPM8O-8,26475
+flamo/optimize/surface.py,sha256=sWy1ImwxUh_QLoY6S68LXBa82_HdWJGplFg2ObtpNGc,26655
 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=a5etio4qfJWwAsHpAk3CXDgEd3LMet-r_956216SO6M,128874
 flamo/processor/system.py,sha256=Hct-o6IgF5NQ2xYbX-1j3st94hMoM8dOgAzle2gjDqU,43145
-flamo-0.2.1.dist-info/METADATA,sha256=sOS3bLK0Cf-EaySS5RxXTtXUilTJ4iHkAbEmnB_VL0w,7825
-flamo-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flamo-0.2.1.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
-flamo-0.2.1.dist-info/RECORD,,
+flamo-0.2.3.dist-info/METADATA,sha256=er-bkIQNPfkYkufZc7bK_pUZR2oFDCAamPzLvL9ED9g,7825
+flamo-0.2.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flamo-0.2.3.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
+flamo-0.2.3.dist-info/RECORD,,