search

mseitzer / pytorch-fid

Compute FID scores with PyTorch.
Apache License 2.0
3.22k stars 497 forks source link

Set number of dataloader workers to a sane default #66

Closed braun-steven closed 3 years ago

braun-steven commented 3 years ago

The prior default was the number of available CPUs which leads to issues for systems with a large number of CPUs (e.g. DGX).

This change introduces an optional argument "--num-workers" which defaults to 8.

Signed-off-by: Steven Lang steven.lang.mz@gmail.com

braun-steven commented 3 years ago

Thanks for your fast response, I've incorporated the requested changes.