ValueError: Can not squeeze dim[1], expected a dimension of 1, got 3 for 'sparse_softmax_cross_entropy_loss

Question

I tried to replace the training and validation data with local images. But when running the training code, it came up with the error :

ValueError: Ca

Answer 1

Changing

loss='sparse_categorical_crossentropy'

to

loss='categorical_crossentropy'

worked for me.