Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "616c3a42cb23161ce2f3c78ed1920f40465c367b"
Unverified Commit 963f8559 authored by Andrei Ivanov's avatar Andrei Ivanov Committed by GitHub
Browse files

Ralaxing the limit on the possible number of num_workers used. (#6440)

parent 9d5b897a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/dataloading/dataloader.py
View file @ 963f8559
...@@ -1227,7 +1227,7 @@ class DataLoader(torch.utils.data.DataLoader): ...@@ -1227,7 +1227,7 @@ class DataLoader(torch.utils.data.DataLoader):
else: else:
node0_cores = list(range(psutil.cpu_count(logical=False))) node0_cores = list(range(psutil.cpu_count(logical=False)))
if len(node0_cores) <= self.num_workers: if len(node0_cores) < self.num_workers:
raise Exception("ERROR: more workers than available cores") raise Exception("ERROR: more workers than available cores")
loader_cores = loader_cores or node0_cores[0 : self.num_workers] loader_cores = loader_cores or node0_cores[0 : self.num_workers]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment