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

Memory leak when returning class_specific_scores #53

Open
sabeenlohawala opened this issue Apr 3, 2024 · 0 comments
Open

Memory leak when returning class_specific_scores #53

sabeenlohawala opened this issue Apr 3, 2024 · 0 comments
Assignees
@sabeenlohawala

Comments

@sabeenlohawala
Copy link
Owner

https://github.com/sabeenlohawala/tissue_labeling/blob/45f1e1ee14491246cbb3c0cd6b17932ef5f91814/TissueLabeling/metrics/metrics.py#L86C9-L86C39

Noticed that GPU memory increases with every batch when class_specific_scores is True and training on the new_kwyk_data. This is not the case when class_specific_scores is False → identify the memory leak
@sabeenlohawala sabeenlohawala self-assigned this Apr 3, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@sabeenlohawala sabeenlohawala

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@sabeenlohawala