Iam getting this error when I\'m fitting a model: ValueError: logits and labels must have the same shape ((None, 1000) vs (None, 1)). I am not sure how to fix it. Below is the c