From ba2d37194aa5a6da1f62547fcbf9f7ee4c50f4ca Mon Sep 17 00:00:00 2001
From: Priya Kasimbeg <kasimbeg@google.com>
Date: Tue, 26 Mar 2024 01:43:25 +0000
Subject: [PATCH] set default eval num workers for pytorch data loaders to 0

---
 submission_runner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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()