Target is out of bounds

[PiotrProzorowski]

I was following the tutorial on Convolutional Neural Networks and when I got to the point of running the train_step function i got this error:
IndexError: Target 5 is out of bounds.
Any ideas why this may be?

It is around this point in the main video:
https://youtu.be/V_xro1bcAuA?si=Cb_a1hVtrHziD7ZG&t=66850

[PiotrProzorowski]

Oh, I found the mistake.
When I created an instance of the model I put the wrong number in the output_shape parameter.

This is the incorrect code:
model_2 = FashionMNISTModel_V2(input_shape=1,
hidden_units=10,
output_shape=1)

This is the correction:
model_2 = FashionMNISTModel_V2(input_shape=1,
hidden_units=10,
output_shape=len(class_names))