reduce num_workers to 2

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jan 26, 2024 · 8a2137d · 8a2137d
1 parent e7c0907
commit 8a2137d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/source/tests/pt/test_saveload_dpa1.py b/source/tests/pt/test_saveload_dpa1.py
@@ -79,7 +79,7 @@ def setUp(self):
             self.dataset,
             sampler=torch.utils.data.RandomSampler(self.dataset),
             batch_size=None,
-            num_workers=8,  # setting to 0 diverges the behavior of its iterator; should be >=1
+            num_workers=2,  # setting to 0 diverges the behavior of its iterator; should be >=1
             drop_last=False,
             pin_memory=True,
         )

diff --git a/source/tests/pt/test_saveload_se_e2_a.py b/source/tests/pt/test_saveload_se_e2_a.py
@@ -79,7 +79,7 @@ def setUp(self):
             self.dataset,
             sampler=torch.utils.data.RandomSampler(self.dataset),
             batch_size=None,
-            num_workers=8,  # setting to 0 diverges the behavior of its iterator; should be >=1
+            num_workers=2,  # setting to 0 diverges the behavior of its iterator; should be >=1
             drop_last=False,
             pin_memory=True,
         )