Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Feature: support multiple GPUS #73

Open
ebartrum opened this issue Jun 17, 2021 · 2 comments
Open

Feature: support multiple GPUS #73

ebartrum opened this issue Jun 17, 2021 · 2 comments
Labels
enhancement New feature or request

Comments

@ebartrum
Copy link

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
Copy link
Owner

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
Copy link
Author

OK, Ill work on a PR!

@mseitzer mseitzer changed the title Any plan to support multiple GPUS? Feature: support multiple GPUS Oct 10, 2021
@mseitzer mseitzer added the enhancement New feature or request label Oct 10, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@ebartrum @mseitzer