C3W2A Assignment - Fail even though passed all unit tests but training stops after epoch 197

Hi Jswat!

The problem is due to a version mismatch! CenterCrop() doesn’t work with grader. All you need to do is slice the matrices directly without the use of CenterCrop() for now.
Take a look at this thread, hope this will clear your issues. If not, let us know.