Tensorflow : logits and labels must have the same first dimension

Question

I am new in tensoflow and I want to adapt the MNIST tutorial https://www.tensorflow.org/tutorials/layers with my own data (images of 40x40). This is my model function :

Answer 1

I resolved it changing from sparse_categorical_crossentropy to categorical_crossentropy and is now running fine.