Commit 59f202ba authored by lukas leufen's avatar lukas leufen 👻
Browse files

halve also data loading in batch creation

parent ba69f14f
Pipeline #101697 failed with stages
in 9 minutes and 54 seconds
......@@ -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:
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment