Hi @damf.383
I guess that your issue is caused by the ‘loss’ function you are using.
I think that in this case the most suitable loss function is
sparse_categorical_crossentropy
because the labels are provided as integers.
Which loss function did you use?
Best regards
Fabio