flamo 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

flamo/auxiliary/reverb.py

@@ -416,7 +416,8 @@ class parallelGFDNAccurateGEQ(parallelFDNAccurateGEQ):
             alias_decay_db=alias_decay_db,
             start_freq=start_freq,
             end_freq=end_freq,
-            device=device
+            device=device,
+            dtype=dtype
         )
         self.n_gains = self.size[0]
         self.size = (self.n_groups * self.size[0],)

flamo/optimize/trainer.py

@@ -54,11 +54,13 @@ class Trainer:
         patience_delta: float = 0.01,
         step_size: int = 50,
         step_factor: float = 0.1,
+        log: bool = True,
         train_dir: str = None,
         device: str = "cpu",
     ):
 
         self.device = device
+        self.log = log
         self.net = net.to(device)
         self.max_epochs = max_epochs
         self.lr = lr
@@ -67,10 +69,10 @@ class Trainer:
         self.min_val_loss = float("inf")
         self.optimizer = torch.optim.Adam(self.net.parameters(), lr=self.lr)
         self.n_loss = 0
-
-        assert os.path.isdir(
-            train_dir
-        ), "The directory specified in train_dir does not exist."
+        if self.log:
+            assert os.path.isdir(
+                train_dir
+            ), "The directory specified in train_dir does not exist."
         self.train_dir = train_dir
 
         self.criterion, self.alpha, self.requires_model = (
@@ -141,7 +143,8 @@ class Trainer:
             self.print_results(epoch, et_epoch - st_epoch)
 
             # save checkpoints
-            self.save_model(epoch)
+            if self.log:
+                self.save_model(epoch)
             if self.early_stop():
                 print("Early stopping at epoch: {}".format(epoch))
                 break

{flamo-0.2.5.dist-info → flamo-0.2.7.dist-info}/METADATA

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

@@ -5,7 +5,7 @@ flamo/auxiliary/__init__.py,sha256=7lVNh8OxHloZ4KPmp-iTUJnUbi8XbuRzGaQ3Z-NKXio,4
 flamo/auxiliary/eq.py,sha256=aI0og6x82r_o3Q7xGu36scdf51e2ARhYmFyJqnEIzQc,7364
 flamo/auxiliary/filterbank.py,sha256=02w8dI8HoNDtKpdVhSJkIkd-h-KNXvZtivf3l4_ozzU,9866
 flamo/auxiliary/minimize.py,sha256=fMTAAAk9yD7Y4luKS4XA1-HTq44xo2opq_dRPRrhlIY,2474
-flamo/auxiliary/reverb.py,sha256=jrNsNUH5B0S-1U-DfPnNVGqpbKCZHfeiFmoCGeiX1QU,34915
+flamo/auxiliary/reverb.py,sha256=4D4minxSW88Vu_xxAYxaU8Z98ag1F4jHfIft4nef0Bg,34940
 flamo/auxiliary/scattering.py,sha256=IV9ZgkMfgEWtQo3IeLJviAXAdjrYI_Tob-H_8Zv8oBA,9729
 flamo/auxiliary/velvet.py,sha256=B4pYEnhaQPkh02pxqiGdAhLRX2g-eWtHezphi0_h4Qs,4201
 flamo/auxiliary/config/config.py,sha256=sZ3XvqwV6KiIc2n8HRtg7YJE3zhc7Vqblbqs-Z0bsKg,2978
@@ -13,12 +13,12 @@ flamo/optimize/__init__.py,sha256=grgxLmQ7m-c9MvRdIejmEAaaajfBwgeaZAv2qjHIvPw,65
 flamo/optimize/dataset.py,sha256=WPvWDhT-U-gFkPaP1UzvFfB2bxlxdDDQ64zQ2-OcbYY,6789
 flamo/optimize/loss.py,sha256=h6EeqjdX5P1SqDBKBavSxV25VBgnYK8tuX91wk6lw_g,33466
 flamo/optimize/surface.py,sha256=sWy1ImwxUh_QLoY6S68LXBa82_HdWJGplFg2ObtpNGc,26655
-flamo/optimize/trainer.py,sha256=he4nUjLC-3RTlxxBIw33r5k8mQfgAGvN1wpPBAWCjVo,12045
+flamo/optimize/trainer.py,sha256=LITPVS87mI6bnq4J6GIXqGb4wW7TKWVXeCu4UQ-csxM,12155
 flamo/optimize/utils.py,sha256=R5-KoZagRho3eykY88pC3UB2mc5SsE4Yv9X-ogskXdA,1610
 flamo/processor/__init__.py,sha256=paGdxGVZgA2VAs0tBwRd0bobzGxeyK79DS7ZGO8drkI,41
 flamo/processor/dsp.py,sha256=Znp_9qjHRb7V0DBaqPWxa9oOlU84CVTjy6h3DcAh-TU,144811
 flamo/processor/system.py,sha256=Hct-o6IgF5NQ2xYbX-1j3st94hMoM8dOgAzle2gjDqU,43145
-flamo-0.2.5.dist-info/METADATA,sha256=ilg6hr1DaZgWg2qq783vj0vd8VNE1DQVsMb3-kKDt24,7825
-flamo-0.2.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flamo-0.2.5.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
-flamo-0.2.5.dist-info/RECORD,,
+flamo-0.2.7.dist-info/METADATA,sha256=WR7SkIg6PT-N6T30zURuCnRt9Lcodgk6RcxG5dghM68,7825
+flamo-0.2.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flamo-0.2.7.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
+flamo-0.2.7.dist-info/RECORD,,