When creating a post, please add:
- Link to the classroom item you are referring to:
I don’t know what I am doing wrong with the week 1 assigment. When the model is trained is taking so long that never ends. It is taking more than 30 minutes to execute and then the google collab runs out of time. I am using at least 3 convolution layers and everything is working out perfect until I need to train the model.
It is part of the Tensorflow Professional Certification. Course 2: Convolutional Neural Netwirks in TensorFlow, week 1. The Assignment is creating the convolutional Neural Networks, a cat and dog classifier. When using the training data: 12500 images for dogs and 12500 images for cats, it never ends.
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)