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

Set torch default device and dtype #25

Open
Markus-Goetz opened this issue Feb 14, 2024 · 1 comment
Open

Set torch default device and dtype #25

Markus-Goetz opened this issue Feb 14, 2024 · 1 comment
Assignees
@Markus-Goetz @MarleneBusch
Labels
maintenance Improve code readability and robustness.
Milestone
ARTIST Release 1.0

Comments

@Markus-Goetz
Copy link
Collaborator

Upon initialization of artist, call torch.set_default_device and torch.set_default_dtype and get rid of explicit dtype and devices within the code base
@Markus-Goetz Markus-Goetz added the maintenance Improve code readability and robustness. label Feb 14, 2024
@kalebphipps kalebphipps removed this from the Parallel Functionality Included milestone Oct 15, 2024
@kalebphipps kalebphipps mentioned this issue Oct 15, 2024
Closed
3 tasks
@kalebphipps kalebphipps added this to the ARTIST Release 1.0 milestone Oct 15, 2024
@MarleneBusch
Copy link
Collaborator

Currently the device is set once in the beginning and is passed down through each function.
We might want to change that and use torch.set_default_device() once.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@Markus-Goetz Markus-Goetz

@MarleneBusch MarleneBusch

Labels
maintenance Improve code readability and robustness.
Projects
None yet
Milestone
ARTIST Release 1.0
Development

No branches or pull requests
3 participants
@Markus-Goetz @kalebphipps @MarleneBusch