https://leapcell.io/blog/understanding-pytorch-dataloader-efficient-data-loading-for-deep-learning