diff --git a/configs/vision/radiology/offline/segmentation/lits_balanced.yaml b/configs/vision/radiology/offline/segmentation/lits_balanced.yaml index b24cf572..28e76c59 100644 --- a/configs/vision/radiology/offline/segmentation/lits_balanced.yaml +++ b/configs/vision/radiology/offline/segmentation/lits_balanced.yaml @@ -7,6 +7,9 @@ trainer: max_steps: &MAX_STEPS ${oc.env:MAX_STEPS, 500000} callbacks: - class_path: eva.callbacks.ConfigurationLogger + - class_path: lightning.pytorch.callbacks.TQDMProgressBar + init_args: + refresh_rate: ${oc.env:TQDM_REFRESH_RATE, 1} - class_path: eva.vision.callbacks.SemanticSegmentationLogger init_args: log_every_n_epochs: 1 diff --git a/configs/vision/radiology/online/segmentation/lits_balanced.yaml b/configs/vision/radiology/online/segmentation/lits_balanced.yaml index 0847de90..e767c097 100644 --- a/configs/vision/radiology/online/segmentation/lits_balanced.yaml +++ b/configs/vision/radiology/online/segmentation/lits_balanced.yaml @@ -8,6 +8,9 @@ trainer: log_every_n_steps: 6 callbacks: - class_path: eva.callbacks.ConfigurationLogger + - class_path: lightning.pytorch.callbacks.TQDMProgressBar + init_args: + refresh_rate: ${oc.env:TQDM_REFRESH_RATE, 1} - class_path: eva.vision.callbacks.SemanticSegmentationLogger init_args: log_every_n_epochs: 1