update warning

mlcommons · Mar 29, 2024 · 28adc86 · 28adc86
1 parent e29a2d4
commit 28adc86
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/submission_runner.py b/submission_runner.py
@@ -154,11 +154,12 @@
 flags.DEFINE_boolean('set_pytorch_max_split_size',
                      False,
                      'If true, set pytorch max_split_size_mb to 256')
-flags.DEFINE_integer('pytorch_eval_num_workers',
-                     0,
-                     'Number of workers for PyTorch evaluation data loaders.'
-                     'WARNING: there is an known bug that results in wrong'
-                     'evals when the number of workers is not equal to 0.')
+flags.DEFINE_integer(
+    'pytorch_eval_num_workers',
+    0,
+    'Number of workers for ImageNet PyTorch evaluation data loaders.'
+    'WARNING: Setting pytorch_eval_num_workers != 0, will result '
+    'in incorrect evals currently, see issues/732.')
 FLAGS = flags.FLAGS
 USE_PYTORCH_DDP, RANK, DEVICE, N_GPUS = pytorch_setup()