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

I have solved this error. The labels were in onehot encoding, so it was in dimension of [,10], rather than [,1]. So I used tf.argmax().