diff --git a/submission_runner.py b/submission_runner.py index 2f8cc9b7b..2945e3fd0 100644 --- 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()