Load mnist how to get the labels?

OK so just indexing will decide on train and validation, and test set is test set. I may not use that for my blitz experiments. Good.