You can use bigger num_workers in dataloader to set how many subprocesses to use for data loading.
pin_memory=True can also help if you’re using gpu.
warning: it’s not working perfectly on windows machine. and you should read dataloader doc for more info about Platform-specific behaviors.