mnist.set(60000, 10000) does not output datasets the right size. I got different lengths in my datasets.