feat: Add option to load entire dataset into memory for faster access

`DataGenerator` now provides an option to load the entire dataset into memory (provided enough memory is available of course). This reduces the time to get the images for a batch during data generation to a negligible amount.

Author: pierluigiferrari