Error message: ValueError: logits and labels must have the same shape ((None, 2) vs (None, 1)) What can be the problem?