Using tf 2.4 and most recent Keras I\'ve subclassed Keras generator from tensorflow.keras.utils.Sequence. Generator forms batches of training data:
batch_x.sh