format

agosztolai · Jun 17, 2024 · 3fd2b94 · 3fd2b94
1 parent f9e2da3
commit 3fd2b94
Showing 1 changed file with 1 addition and 3 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(