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