The 20 iterations thing was just to mimic using a batch of 20.
I’ve messed around with the learning rates and still don’t seem to make any progress.
So I’m still kind of confused on where the issue is.