From e84db37680f70cf52d662899affd816ebe7c8742 Mon Sep 17 00:00:00 2001 From: Cadene Date: Thu, 2 May 2024 20:59:53 +0000 Subject: [PATCH] try lowering number of workers --- tests/test_datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index ea72f682..236a7032 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -115,7 +115,7 @@ def test_compute_stats_on_xarm(): # get all frames from the dataset in the same dtype and range as during compute_stats dataloader = torch.utils.data.DataLoader( dataset, - num_workers=8, + num_workers=1, batch_size=len(dataset), shuffle=False, )