Loading batch data while training is slow when train on large dataset. Could we speed up data loading?