Merge pull request #723 from mlcommons/num_workers_fix

set default eval num workers for pytorch data loaders to 0
mlcommons · Mar 26, 2024 · ec31805 · ec31805
2 parents c1b6959 + ba2d371
commit ec31805
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/submission_runner.py b/submission_runner.py
@@ -155,7 +155,7 @@
                      False,
                      'If true, set pytorch max_split_size_mb to 256')
 flags.DEFINE_integer('pytorch_eval_num_workers',
-                     4,
+                     0,
                      'Number of workers for PyTorch evaluation data loaders.')
 FLAGS = flags.FLAGS
 USE_PYTORCH_DDP, RANK, DEVICE, N_GPUS = pytorch_setup()