From 4f12ebafbad292867dd4cd79f6360143b4af2110 Mon Sep 17 00:00:00 2001 From: Egor Date: Thu, 8 Feb 2024 14:22:52 +0100 Subject: [PATCH] Disabled background batch processing --- dl_bench/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/dl_bench/utils.py b/dl_bench/utils.py index af9bba7..28776e5 100644 --- a/dl_bench/utils.py +++ b/dl_bench/utils.py @@ -34,10 +34,10 @@ def get_inf_loaders(n, in_shape, batch_size, device: str): ds = RandomInfDataset(n, in_shape) train_loader = DataLoader( - ds, batch_size=batch_size, shuffle=True, num_workers=4, pin_memory=pin_memory + ds, batch_size=batch_size, shuffle=True, num_workers=0, pin_memory=pin_memory ) test_loader = DataLoader( - ds, batch_size=batch_size, shuffle=False, num_workers=4, pin_memory=pin_memory + ds, batch_size=batch_size, shuffle=False, num_workers=0, pin_memory=pin_memory ) return train_loader, test_loader @@ -363,7 +363,7 @@ def inference(self, backend: Backend): self.dataset, batch_size=self.batch_size, shuffle=False, - num_workers=4, + num_workers=0, pin_memory=backend.device_name == "cuda", )