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

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

It could be be, depending on the dataset size!

1 Like

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

1 Like

The dataset is big thats for sure!

1 Like

Thanks.

1 Like

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

  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