pytorch-fid icon indicating copy to clipboard operation
pytorch-fid copied to clipboard

Published 20 hours ago • verified publisher icon mseitzer

Feature: support multiple GPUS

Open ebartrum opened this issue 3 years ago • 2 comments

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?

ebartrum avatar Jun 17 '21 10:06 ebartrum

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.

mseitzer avatar Jun 23 '21 16:06 mseitzer

OK, Ill work on a PR!

ebartrum avatar Jun 23 '21 16:06 ebartrum