예제5.9 오류

[sjleergb]

예제5.9 실행하면
in : history = model.fit(train_X,train_Y,epochs=25, batch_size=32, validation_split=0.25)
out :
InvalidArgumentError: logits and labels must be broadcastable: logits_size=[32,2] labels_size=[512,2]
[[node loss/dense_7_loss/softmax_cross_entropy_with_logits (defined at :1) ]] [Op:__inference_distributed_function_2129]

이렇게 오류가 발생합니다. 이전 예제까지는 수월히 됩니다. 어찌 해야 하나요?