I\'m training a CNN and am applying data augmentation through Keras layers by defining:
data_augmentation = keras.Sequential([ layers.experimental.preproces