Skip to content

Conversation

@gleize
Copy link
Contributor

@gleize gleize commented Mar 31, 2024

This is a fix for a memory leak (of a few megabytes per call).
The line

tmp_x[1:] = tmp_x[1:] + tmp_feats

creates a reference cycle that the garbage collector is unable to release.

Here is the active memory use before and after the fix (ran 3 epochs of 10 steps).
Screenshot 2024-03-31 at 10 37 58 AM

@shepnerd shepnerd merged commit 8c729cb into OpenGVLab:main Mar 31, 2024
Andy1621 pushed a commit that referenced this pull request Apr 13, 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