logits and labels must be broadcastable: logits_size=[64,5] labels_size=[64,11] [[{{node categorical_crossentropy/softmax_cross_entropy_with_logits}}]] [Op:__inference_train_function_9008]

ERROR

 

상황

Retrain MobileNet V2 classifier for the Edge TPU (TF2) Code 사용 중에 DataSet 변경 후 Train 진행

 

해결

OutputLayer에서 units 값을 자신의 Class 개수로 변경