Number of workers should be the number of cpus. The preprocessing should be done in the dataset class not data loader.
Number of workers should be the number of cpus. The preprocessing should be done in the dataset class not data loader.