Skip to content

Commit 28adc86

Browse files
committed
update warning
1 parent e29a2d4 commit 28adc86

1 file changed

Lines changed: 6 additions & 5 deletions

File tree

submission_runner.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -154,11 +154,12 @@
154154
flags.DEFINE_boolean('set_pytorch_max_split_size',
155155
False,
156156
'If true, set pytorch max_split_size_mb to 256')
157-
flags.DEFINE_integer('pytorch_eval_num_workers',
158-
0,
159-
'Number of workers for PyTorch evaluation data loaders.'
160-
'WARNING: there is an known bug that results in wrong'
161-
'evals when the number of workers is not equal to 0.')
157+
flags.DEFINE_integer(
158+
'pytorch_eval_num_workers',
159+
0,
160+
'Number of workers for ImageNet PyTorch evaluation data loaders.'
161+
'WARNING: Setting pytorch_eval_num_workers != 0, will result '
162+
'in incorrect evals currently, see issues/732.')
162163
FLAGS = flags.FLAGS
163164
USE_PYTORCH_DDP, RANK, DEVICE, N_GPUS = pytorch_setup()
164165

0 commit comments

Comments
 (0)