I\'m adding data augmentation to my tensorflow model like so:
data_augmentation = keras.Sequential([ layers.experimental.preprocessing.RandomRotation(factor