From 5a3d807ec15bda749902c2f97002af0c5279d5a7 Mon Sep 17 00:00:00 2001 From: Harsha Date: Mon, 18 Mar 2024 13:58:19 -0400 Subject: [PATCH] resolved https://github.com/neuronets/nobrainer/issues/296 --- nobrainer/processing/segmentation.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/nobrainer/processing/segmentation.py b/nobrainer/processing/segmentation.py index faf1a7b7..63648d90 100644 --- 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