Train function gives an error plz look into it

#1

i am attaching googgle colab notebook

https://colab.research.google.com/drive/1gPvaw6oby_ha2n0OxKZNbk-CVlrzISHa?usp=sharing

#train our model#

history = model.fit_generator(
train_generator,
steps_per_epoch = 7,
epochs = 20

)

InvalidArgumentError: logits and labels must be broadcastable: logits_size=[32,10] labels_size=[32,4]
[[node categorical_crossentropy/softmax_cross_entropy_with_logits (defined at :6) ]] [Op:__inference_train_function_988]

#2

hey @Aashi-Agarwal-2380268668952871 ,
Its a coding error.
In your model you have stated the output to be of shape (10,) where as your labels are of shape (4,).
So just kindly correct that , it will work find afterwards.