BUG FIX: test set should be 50k examples (not 40k)
Also, make sure we compute standardization statistics on the first 10k examples
(training set), before we start playing with the data (and extracting subsets,
etc.)