Fixed a bug in getting the datasets and rescaling. Unfortunately (and surprisingly) the files are too large to fit into memory. We'll have to save multiple rescaled files to disk and train using one file at a time in memory