Fix buffers not being freed

vislearn / FrEIA-community

Community-driven effort to improve and extend FrEIA

0 stars 0 forks source link

Fix buffers not being freed #7

Open wapu opened 3 years ago

wapu commented 3 years ago

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()

fdraxler commented 3 years ago

I guess this could be merged with #8

ardizzone commented 3 years ago

Also see https://github.com/VLL-HD/FrEIA/issues/10 (can be closed once this is implemented)