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

Out of memory error while training the model #71

Open
Haneesha123 opened this issue Apr 4, 2021 · 1 comment
Open

Out of memory error while training the model #71

Haneesha123 opened this issue Apr 4, 2021 · 1 comment

Comments

@Haneesha123
Copy link

Hello,
When I try to train your model, I am getting the Runtime error:CUDA: out of memory error. Can you suggest me anything to get rid of this error?

Thank you.
@layumi
Copy link
Contributor

layumi commented May 26, 2021

Hi @Haneesha123
You may try fp16 to save the GPU memory cost.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@layumi @Haneesha123