resolved #296

neuronets · Mar 18, 2024 · 5a3d807 · 5a3d807
1 parent b3e2a8a
commit 5a3d807
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nobrainer/processing/segmentation.py b/nobrainer/processing/segmentation.py
@@ -40,6 +40,7 @@ def fit(
         loss=losses.dice,
         metrics=metrics.dice,
         callbacks=None,
+        verbose=1,
     ):
         """Train a segmentation model"""
         # TODO: check validity of datasets
@@ -100,6 +101,7 @@ def _compile():
             if dataset_validate
             else None,
             callbacks=callbacks,
+            verbose=verbose
         )
 
         return self