I\'m trying to train a simple model over some picture data that belongs to 10 classes. The images are in B/W format (not gray scale), I\'m using the image_dataset_from_direc
