Feature: support multiple GPUS

mseitzer / pytorch-fid

Compute FID scores with PyTorch.

Apache License 2.0

3.22k stars 497 forks source link

Feature: support multiple GPUS #73

Open ebartrum opened 3 years ago

ebartrum commented 3 years ago

Many GAN repos call calculate_fid_given_paths during training to monitor performance. We can train with n GPUS but have to pause all and run the validation with a single GPU, which is a speed bottleneck. Could it be distributed?

mseitzer commented 3 years ago

This sounds like a good idea, but I don't know when I will have time to implement it. I would welcome PRs for this.

ebartrum commented 3 years ago

OK, Ill work on a PR!