Skip to content

Conversation

aovoc
Copy link
Contributor

@aovoc aovoc commented Aug 10, 2019

Fix out of memory error by loading the checkpoint onto the CPU.

Fixing out of memory error by loading the checkpoint onto the CPU.
Copy link
Collaborator

@rwightman rwightman left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks

@rwightman rwightman merged commit f937e2a into huggingface:master Aug 10, 2019
@aovoc aovoc deleted the patch-1 branch August 11, 2019 05:57
guoriyue pushed a commit to guoriyue/pytorch-image-models that referenced this pull request May 24, 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.

2 participants