set default eval num workers for pytorch data loaders to 0

mlcommons · Mar 26, 2024 · ba2d371 · ba2d371
1 parent c1b6959
commit ba2d371
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()