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

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.6.dist-info}/METADATA

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

@@ -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.6.dist-info/METADATA,sha256=Rq2FeB7cAFHVtNtRIz0NS6wOAAizFuwTr9UwJW6JiS8,7825
+flamo-0.2.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flamo-0.2.6.dist-info/licenses/LICENSE,sha256=smMocRH7xdPT5RvFNqSLtbSNzohXJM5G_rX1Qaej6vg,1120
+flamo-0.2.6.dist-info/RECORD,,