Sorry for this amateur question but I\'ve been stuck for days.
I\'m loading the MNIST data into google colab, which I can see is about 60,000 samples
But for
