I used to familiar with the Dataset class in pytorch, when calling torch.utils.data.DataLoader(), you can set num_workers= xxx to enable multi-process to load the data(according