GDSS icon indicating copy to clipboard operation
GDSS copied to clipboard

Published 20 hours ago • verified publisher icon harryjo97

num_workers > 0

Open pavelbatyr opened this issue 2 years ago • 2 comments
trafficstars

Hi, and thanks for the code!

I suggest using num_workers > 0 in the dataloaders. It can speed up training significantly. Using 4 workers makes training 30% faster in my case.

pavelbatyr avatar Jun 14 '23 12:06 pavelbatyr

Hi,

Thank you for the suggestion! We would add the num_workers feature to the following line. https://github.com/harryjo97/GDSS/blob/4d96334fd0d07577f9891e9d5e81dae4d64a92fd/utils/data_loader.py#L12

harryjo97 avatar Jun 14 '23 12:06 harryjo97

Thanks! There's also a dataloader for molecules in utils/data_loader_mol.py

pavelbatyr avatar Jun 14 '23 13:06 pavelbatyr