Error after submitting MNIST assignment

Hi All,
I’m getting following errors and not sure why it’s happening. It running fine before submitting but after grading
its showing errors.

I’ve submitted multiple times and different section is showing error each time. pls guide.

submission -2:

code section:

  1. reshape_and_normalize

  2. Callback

  3. Model

  4. Training

Thanks and Regards,

Grader feedback to optimize the solution points to making the model size smaller. Please try again.
Don’t forget to remove code from your post.