C4w4 train_step

Course Q&A Deep Learning Specialization Convolutional Neural Networks
1

For people struggling to figure it out, please clean your output and run your cells in order!!
took me ages to realise that this was my problem.

2

This a very good point.