Using Tensorflow Dataset from_generator() to create multi Input/Output with Custom Generator and ImageDataGenerator

Question

I am trying to scale up my model which uses a "cluster loss" extension, the implementation works so far on MNIST, but I would like to benefit from data augmentatio