From 8d369d63a52ef9e6ceaaeb07b7ef66f711f1cfa2 Mon Sep 17 00:00:00 2001
From: Felix Kleinert <f.kleinert@fz-juelich.de>
Date: Tue, 9 Aug 2022 17:28:00 +0200
Subject: [PATCH] set counter for ens pred per station
---
mlair/run_modules/post_processing.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py
index 3317af7b..3ad8c4f5 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))
--
GitLab