Glad that you could make it work.
Oh, I just realized that persistent_workers was not available for pytorch 1.6.0 and appears in the docs since 1.7.0. As idist.auto_dataloader just pass through the args and pytorch DataLoader does not have them, that’s why there is an issue.