Why do you think it is not working in the current DataLoader implementation?
The num_workers argument will specify the number of processes, which will load and process each batch in the background, while the main training loop is busy.
1 Like