From 59f202ba04fc394395ce2bb050046a4eb39232b3 Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Tue, 31 May 2022 15:11:27 +0200
Subject: [PATCH] halve also data loading in batch creation
---
mlair/data_handler/iterator.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mlair/data_handler/iterator.py b/mlair/data_handler/iterator.py
index e353f84d..3fc25a90 100644
--- a/mlair/data_handler/iterator.py
+++ b/mlair/data_handler/iterator.py
@@ -144,8 +144,8 @@ class KerasIterator(keras.utils.Sequence):
mod_rank = self._get_model_rank()
for data in self._collection:
logging.debug(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)]
+ X, _Y = data.get_data(upsampling=self.upsampling)
+ Y = [_Y[0] for _ in range(mod_rank)]
if self.upsampling:
X, Y = self._permute_data(X, Y)
if remaining is not None:
--
GitLab