Merge branch 'main' of github.com:agosztolai/MARBLE

agosztolai · Jun 17, 2024 · adeaa83 · adeaa83
2 parents f030fcb + f8c88cc
commit adeaa83
Show file tree

Hide file tree

Showing 2 changed files with 77 additions and 2,056 deletions.
diff --git a/MARBLE/main.py b/MARBLE/main.py
@@ -378,9 +378,7 @@ def fit(self, data, outdir=None, verbose=False):
             train_loss, optimizer = self.batch_loss(
                 data, train_loader, train=True, verbose=verbose, optimizer=optimizer
             )
-            val_loss, _ = self.batch_loss(
-                data, val_loader, verbose=verbose
-            )
+            val_loss, _ = self.batch_loss(data, val_loader, verbose=verbose)
             scheduler.step(train_loss)
 
             print(

diff --git a/examples/RNN/RNN.ipynb b/examples/RNN/RNN.ipynb