Skip to content
Snippets Groups Projects
Commit fe665a8e authored by leufen1's avatar leufen1
Browse files

add info logging of batches

parent c34efef0
Branches
Tags
4 merge requests!430update recent developments,!413update release branch,!412Resolve "release v2.0.0",!377Resolve "add log if batch creation fails"
Pipeline #86989 passed
......@@ -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:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment