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

load ckpt to cpu #428

Merged
merged 2 commits into from
Oct 1, 2024
Merged

load ckpt to cpu #428

merged 2 commits into from
Oct 1, 2024

Conversation

benjijamorris
Copy link
Contributor

What does this PR do?

When loading weights from checkpoint, default to cpu. Once the weights are loaded, lightning will move the model to the appropriate device.

Before submitting

  • Did you make sure title is self-explanatory and the description concisely explains the PR?
  • Did you make sure your PR does only one thing, instead of bundling different changes together?
  • Did you list all the breaking changes introduced by this pull request?
  • Did you test your PR locally with pytest command?
  • Did you run pre-commit hooks with pre-commit run -a command?

Did you have fun?

Make sure you had fun coding 🙃

@benjijamorris benjijamorris merged commit 49b0bc0 into main Oct 1, 2024
4 of 6 checks passed
@benjijamorris benjijamorris deleted the load_ckpt_to_cpu branch October 1, 2024 18:52
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@ritvikvasan ritvikvasan ritvikvasan approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@benjijamorris @ritvikvasan