From 9cf2856ceaa7a6a528b0cc6d95a80810b5293c80 Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Fri, 31 May 2024 11:56:28 +0000 Subject: [PATCH] fix --- tests/test_policies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_policies.py b/tests/test_policies.py index 22d9c294..c099bef0 100644 --- a/tests/test_policies.py +++ b/tests/test_policies.py @@ -140,7 +140,7 @@ def test_policy(env_name, policy_name, extra_overrides): dataloader = torch.utils.data.DataLoader( dataset, - num_workers=4, + num_workers=0, batch_size=2, shuffle=True, pin_memory=DEVICE != "cpu",