C2 Week 2 - trainning the model takes hours. Reasonable?

Course Q&A TensorFlow Developer Professional Convolutional Neural Networks in TensorFlow
1

The colab doesn’t allow me GPU T4 option, just GPU.
Is it reasonable that each epoch takes about 18 minutes?

I tried different batch sizes.

1 Like
2

It could be be, depending on the dataset size!

1 Like
3

This sepcific exercise is with dogs and cats full dataset, 25k images, with augmentation. Input size 150*150.

1 Like
4

The dataset is big thats for sure!

1 Like
5

Thanks.

1 Like
6

The reason for slowness is ImageDataGenerator (see this). Here’s my recommendation:

  1. Use image_dataset_from_directory to iterate on model architecture.
  2. Swap image_dataset_from_directory with ImageDataGenerator after finalizing the model architecture and pay the price once to please the grader.
1 Like