I've successfully replicated your code using the versions that you specified, and it's functioning . The model is training with defined epochs. It looks like the problem might be related to how the image data generator is processing the data from the directory. Please check the provided path and verify that your main directory contains sub-folders, each representing a different class, matching the total number of classes in your dataset.

Please refer to this gist