improve implementation of num_workers input for DataLoaders

[sophie22]

num_workers in pytorch may be set to 0 to use all available workers, but MLOps uses the num_workers value directly to set the max_workers, causing the below error