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

Fix buffers not being freed #7

Open
wapu opened this issue Feb 8, 2021 · 2 comments
Open

Fix buffers not being freed #7

wapu opened this issue Feb 8, 2021 · 2 comments
Assignees

Comments

@wapu
Copy link
Collaborator

wapu commented Feb 8, 2021

We need to fix the issue with ReversibleGraphNet._buffers (i.e. edges on the computation graph)
not freeing once they are no longer needed, and also being included in state_dict()

@wapu wapu changed the title Fix ._buffers not being freed Fix buffers not being freed Feb 8, 2021
@fdraxler
Copy link
Collaborator

fdraxler commented Feb 9, 2021

I guess this could be merged with #8

@fdraxler fdraxler mentioned this issue Feb 9, 2021
@ardizzone
Copy link
Member

ardizzone commented Feb 9, 2021

Also see
vislearn/FrEIA#10
(can be closed once this is implemented)

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

No branches or pull requests

3 participants