The question seems to be rather Keras/TF-specific and I think you would get the best support on e.g. StackOverflow. I’m unfortunately not deeply familiar with Keras and don’t know, how the data loading (lazy loading) can be implemented.