From 732a020dd4609485e7435c75310abcb4956c54ea Mon Sep 17 00:00:00 2001 From: Andreas Moldskred Date: Mon, 1 Apr 2024 21:35:44 -0700 Subject: [PATCH] Ensure worker has access to all cpus --- utils/dataloaders.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utils/dataloaders.py b/utils/dataloaders.py index 776042999..d7ccf4972 100644 --- a/utils/dataloaders.py +++ b/utils/dataloaders.py @@ -93,6 +93,7 @@ def seed_worker(worker_id): worker_seed = torch.initial_seed() % 2 ** 32 np.random.seed(worker_seed) random.seed(worker_seed) + os.sched_setaffinity(0, range(os.cpu_count())) def create_dataloader(path,