Merge pull request #26 from agosztolai/fix_notebook

Fix notebook corrupted
agosztolai · Jun 17, 2024 · f8c88cc · f8c88cc
2 parents 73297ca + 3fd2b94
commit f8c88cc
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