C3W2_Assignment: ValueError

[Marco_Stallmann]

With

[code removed - moderator]

I am getting:

ValueError: logits and labels must have the same shape, received ((None, 5) vs (None, 1)).

Whats wrong here?

[code removed - moderator]

[balaji.ambresh]

Please fix the model loss function.

[Kai_Strasdas]

I have the same problem:
ValueError: logits and labels must have the same shape, received ((None, None, None, 1) vs (None,)).

[balaji.ambresh]

Number of units in the output layer must equal the number of classes you’re trying to predict.

For this assignment, the output shape should be (None, # classes) where None refers to the batch size.