Index Out Of Range Tensorflow Keras With Model.fit()

I am following a blog post from Keras (https://blog.keras.io/building-powerful-image-classification-models-using-very-little-data.html) and have trouble executing the script (origi

Solution 1:

This line looks strange:

  model.add(Flatten(input_shape=train_data.shape[1:]))

Try changing to:

  model.add(Flatten(input_shape=train_data.shape))