Please confirm the following:
- Using a GPU for training the model.
- No augmentations are performed on the train dataset.
Training time for this assignment is influenced by 2 factors:
- Model architecture: Look for the number of parameters the model is training on using
model.summary(). One way to speed up training is to reduce the number of parameters (by using more conv* and *pool layers) since the learnable parameters on a Dense layer goes down. - (Probably out of your hands in this assignment but is good to know) Stop relying on
ImageDataGenerator(read this)