Load faster

[alfredplpl]

The code seems that the image loading method is slow.
For example, I think the code should load learning images in parallel.
Current:

denoising-diffusion-pytorch/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py

Line 601 in f900f40

self.dl = cycle(data.DataLoader(self.ds, batch_size = train_batch_size, shuffle=True, pin_memory=True))

After:
self.dl = cycle(data.DataLoader(self.ds, batch_size = train_batch_size, num_workers=os.cpu_count(), shuffle=True, pin_memory=True))

May I make an pull request?

[alfredplpl]

Thank you for your coding.