thank you mate. I was thinking about that for a while. but then I cannot take advantage of multi processing like I can with num_workers > 0 can I?
Also I did not yet completely understand the use case of TensorDataset, would you care to elaborate a little?