Loading a large dataset and training them

[ptrblck]

You could profile your code and try to isolate, if the data loading is the bottleneck.
If that’s the case, this post might be helpful to further optimizer the loading.