How to have a fluent dataloader?

2

You could use the new persistent_workers argument in the DataLoader construction, which will not shutdown the workers between epochs.