search

w86763777 / pytorch-image-generation-metrics

Pytorch implementation of common image generation metrics.
Apache License 2.0
143 stars 16 forks source link

fix: argument of `DataLoader` #4

Closed david20571015 closed 1 year ago

david20571015 commented 1 year ago

Set the argument num_workers from 4 to the args.num_workers,

w86763777 commented 1 year ago

Thanks for your pull request. I'll accpet this merge first.

In the future, this parameter will be changed to num_workers=min(args.num_workers, torch.get_num_threads()), which can further avoid warnings on different hardware equipments.