From 77fb8ee0823e7a5b99ea7b9a07a439d1e108e30d Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 26 Jun 2020 19:30:09 -0700 Subject: [PATCH] refactor dataloader --- utils/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index 00f2338..aee891c 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -52,7 +52,7 @@ def create_dataloader(path, imgsz, batch_size, stride, opt, hyp=None, augment=Fa pad=pad) batch_size = min(batch_size, len(dataset)) - nw = min([os.cpu_count(), batch_size if batch_size > 1 else 0, 0]) # number of workers + nw = min([os.cpu_count(), batch_size if batch_size > 1 else 0, 8]) # number of workers dataloader = torch.utils.data.DataLoader(dataset, batch_size=batch_size, num_workers=nw,