diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py index 3317af7b0c054e5f4aad691efd3585764da0f7c3..3ad8c4f56525124e4c31eced16317726e11af48d 100644 --- a/mlair/run_modules/post_processing.py +++ b/mlair/run_modules/post_processing.py @@ -736,7 +736,7 @@ class PostProcessing(RunEnvironment): nn_output = self.model.model(input_data) nn_output = nn_output.numpy() else: - with TimeTracking(name=f"Create ensemble predictions for {data.__repr__()} ({subset_type})"): + with TimeTracking(name=f"Create ensemble predictions for {data.__repr__()} ({subset_type}), {i+1}/{len(subset)}"): ens_collector = [] for r in range(self.num_realizations): ens_collector.append(self.model.model(input_data))