Convolutional Neural Networks in TensorFlow week 1 Assigment

Please confirm the following:

  1. Using a GPU for training the model.
  2. No augmentations are performed on the train dataset.

Training time for this assignment is influenced by 2 factors:

  1. 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.
  2. (Probably out of your hands in this assignment but is good to know) Stop relying on ImageDataGenerator (read this)