amend

pytorch · Mar 5, 2024 · a92de2c · a92de2c
1 parent 8b6a038
commit a92de2c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_libs.py b/test/test_libs.py
@@ -2626,7 +2626,7 @@ def test_atari_preproc(self, dataset_id):
             slice_len=None,
             num_slices=8,
             batch_size=64,
-            num_procs=os.cpu_count(),
+            num_procs=max(0, os.cpu_count() - 4),
         )
 
         t = Compose(
@@ -2638,7 +2638,7 @@ def test_atari_preproc(self, dataset_id):
         def preproc(data):
             return t(data)
 
-        dataset.preprocess(preproc, num_workers=max(1, os.cpu_count()-2), num_chunks=1000, mp_start_method="fork", pbar=True)
+        dataset.preprocess(preproc, num_workers=max(1, os.cpu_count()-4), num_chunks=1000, mp_start_method="fork", pbar=True)
         print(dataset)