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

add info logging of batches

parent c34efef0
No related branches found
No related tags found
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 ...@@ -9,6 +9,7 @@ import math
import os import os
import shutil import shutil
import pickle import pickle
import logging
import dill import dill
from typing import Tuple, List from typing import Tuple, List
...@@ -142,6 +143,7 @@ class KerasIterator(keras.utils.Sequence): ...@@ -142,6 +143,7 @@ class KerasIterator(keras.utils.Sequence):
remaining = None remaining = None
mod_rank = self._get_model_rank() mod_rank = self._get_model_rank()
for data in self._collection: for data in self._collection:
logging.info(f"prepare batches for {str(data)}")
X = data.get_X(upsampling=self.upsampling) X = data.get_X(upsampling=self.upsampling)
Y = [data.get_Y(upsampling=self.upsampling)[0] for _ in range(mod_rank)] Y = [data.get_Y(upsampling=self.upsampling)[0] for _ in range(mod_rank)]
if self.upsampling: if self.upsampling:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment