ValueError: Arguments ‘target’ and ‘output’ must have the same shape. Received: target.shape = (None, 3), output.shape=(None, 2) I was trying to train the model.