diff --git a/submission_runner.py b/submission_runner.py index 40eb8cd58..87b2703e1 100644 --- 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()