diff --git a/data/training/train.py b/data/training/train.py index 6d76019..1d15968 100644 --- a/data/training/train.py +++ b/data/training/train.py @@ -138,7 +138,7 @@ def train_model(): benchmark=True, precision="16-mixed", logger=False, - enable_progress_bar=False, # Disable all progress bars + enable_progress_bar=True, # Disable all progress bars enable_model_summary=False, # Disable model summary )