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.

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

image segment memory fix #1341

Merged
merged 1 commit into from Apr 27, 2022
Merged

image segment memory fix #1341

merged 1 commit into from Apr 27, 2022

Conversation

JKL98ISR
Copy link
Member

closes #1148

having tensors with grads cached caused major memory leak

@JKL98ISR JKL98ISR self-assigned this Apr 27, 2022
@JKL98ISR JKL98ISR requested a review from a team as a code owner April 27, 2022 12:52
Copy link
Contributor

@matanper matanper left a comment

Choose a reason for hiding this comment

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

good catch

Copy link
Member

@ItayGabbay ItayGabbay left a comment

Choose a reason for hiding this comment

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

Wow

@JKL98ISR JKL98ISR enabled auto-merge (squash) April 27, 2022 13:08
@JKL98ISR JKL98ISR merged commit e679b25 into main Apr 27, 2022
@delete-merged-branch delete-merged-branch bot deleted the image_segment_mem_fix branch April 27, 2022 13:42
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.

[BUG] [Vision] running ImageSegmentPerformance raises out of memory in both gpu and cpu devices
3 participants