Tensorflow 2.3. How to change the batch for each epoch?

Question

I would like to train with a different custom image augmentation during each epoch in the training.

• The wrong solution would be to save the augmented images, an