From fe665a8e36d9efdc319408f192731713612ca58e Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Mon, 20 Dec 2021 10:30:51 +0100
Subject: [PATCH] add info logging of batches
---
mlair/data_handler/iterator.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/mlair/data_handler/iterator.py b/mlair/data_handler/iterator.py
index f2e3b689..8c7cc6c5 100644
--- a/mlair/data_handler/iterator.py
+++ b/mlair/data_handler/iterator.py
@@ -9,6 +9,7 @@ import math
import os
import shutil
import pickle
+import logging
import dill
from typing import Tuple, List
@@ -142,6 +143,7 @@ class KerasIterator(keras.utils.Sequence):
remaining = None
mod_rank = self._get_model_rank()
for data in self._collection:
+ logging.info(f"prepare batches for {str(data)}")
X = data.get_X(upsampling=self.upsampling)
Y = [data.get_Y(upsampling=self.upsampling)[0] for _ in range(mod_rank)]
if self.upsampling:
--
GitLab