C4w4 train_step

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.

1 Like

This a very good point.

1 Like