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.

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

fix: pass on device to DataParallel; line breaks. #65

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

janfb
Copy link

@janfb janfb commented Jul 22, 2024

fixes #60

Changes:

  • pass on device to DataParallel
  • add a minimal test for fid calculation with different devices
  • add docstrings and types for fid.py

Note that I added docstrings and types for all functions in fid.py. Given that so many people are using the library I thought it would be helpful to give a bit more context for the users. Do let me know though, if you prefer not having theses changes, or having them in a separate PR.

Best,
Jan

@janfb janfb changed the title fix: pass on device to DataParallel; like breaks. fix: pass on device to DataParallel; line breaks. Jul 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

cuda device mismatch in DataParallel when not using cuda:0
1 participant