ok your msg is bit confusing, can you first tell me which one did you use and which grade function cell you used?
check this thread
Also make sure your model follows the architecture instructions give in the assignment, like using the number layers and then dense output layers.
Also confirm what loss you have used categorical crossentropy or sparse categorical crossentropy??
remember here classes are divided into labels, so sparse would be the choice right?