Avoid loading entire dataset before training

[DarnellGranberry]

Training on large datasets raises OOM error before training. Need some alternative such as storing in normal RAM or loading on-the-fly.