I load my datasets with keras.utils.image_dataset_from_directory(). Here's the code:train_ds = im