Slight edit to refelct steps required to fully process dataset to 16 sample GPU batches.