Code Cell UNQ_C5: Function 'train_step' is incorrect

All tests are passed
#Week4 #lab2 Convolutional Neural Networks
Code Cell UNQ_C5: Function ‘train_step’ is incorrect. Check implementation.
If you see many functions being marked as incorrect, try to trace back your steps & identify if there is an incorrect function that is being used in other steps.
This dependency may be the cause of the errors.

The grader uses different tests of your code than those in the notebook.

Passing the notebook’s tests does not mean your code is perfect.