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