Fix "RuntimeError: CUDA Error: out of memory"#39
Open
icekang wants to merge 1 commit intoarnoweng:masterfrom
Open
Fix "RuntimeError: CUDA Error: out of memory"#39icekang wants to merge 1 commit intoarnoweng:masterfrom
icekang wants to merge 1 commit intoarnoweng:masterfrom
Conversation
|
@icekang Thank you!! Changes worked fine and saved me debugging time. |
Author
|
you are welcome :) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Problem
According to this issue
I also forked this repo and tried running on my Colab project. The same problem arose:
RuntimeError: CUDA Error: out of memory.Solution
As far as my knowledge, the problem happened because some section did not require
gradyet still did anyway. Thus,with no_grad()should be presented.