Stuck on Error

[parulkuls26]

After running the below code

# Fitting the model and running the model for 10 epochs history4 = model4.fit( train_images, train_labels, epochs = 10, callbacks = callbacks, batch_size = 32, validation_split = 0.2, verbose = 1 )

I am getting this error

"ValueError: Arguments target and output must have the same shape. Received: target.shape=(None, 2), output.shape=(None, 256)"

can someone help?

[Larz60+]

Please show your code. Use bbcode tags.