Data class holds all samples in memory #2
Description
This won't scale much longer since if I add any more training data it won't all fit in memory. In this project they have a list of all the training images and only load what is required for each training batch.
My current way might be a bit quicker to train since the IO is front loaded but eventually something like this might be required.