From 69eb85ea6365b9dc0676811f3cd06c4ee6c2c09d Mon Sep 17 00:00:00 2001 From: LorenzLamm Date: Sat, 21 Sep 2024 19:25:32 +0200 Subject: [PATCH] add on the ly dataloading to training summary --- .../segmentation/training/training_param_summary.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/membrain_seg/segmentation/training/training_param_summary.py b/src/membrain_seg/segmentation/training/training_param_summary.py index 67277c8..1837811 100644 --- a/src/membrain_seg/segmentation/training/training_param_summary.py +++ b/src/membrain_seg/segmentation/training/training_param_summary.py @@ -3,6 +3,7 @@ def print_training_parameters( log_dir: str = "logs/", batch_size: int = 2, num_workers: int = 8, + on_the_fly_dataloading: bool = False, max_epochs: int = 1000, aug_prob_to_one: bool = False, use_deep_supervision: bool = False, @@ -25,6 +26,8 @@ def print_training_parameters( Number of samples per batch of input data. num_workers : int, optional Number of subprocesses to use for data loading. + on_the_fly_dataloading : bool, optional + If True, data is loaded on the fly. max_epochs : int, optional Maximum number of epochs to train for. aug_prob_to_one : bool, optional @@ -68,6 +71,12 @@ def print_training_parameters( "loading.".format(num_workers) ) print("————————————————————————————————————————————————————————") + on_the_fly_status = "Enabled" if on_the_fly_dataloading else "Disabled" + print( + "On-the-Fly Data Loading:\n {} \n If enabled, data is loaded on " + "the fly.".format(on_the_fly_status) + ) + print("————————————————————————————————————————————————————————") print(f"Max Epochs:\n {max_epochs} \n Maximum number of training epochs.") print("————————————————————————————————————————————————————————") aug_status = "Enabled" if aug_prob_to_one else "Disabled"