yo_its_me

This issue stems from the fact that the registered `forward_hooks` (they are used to grab feature maps from the hidden layers) are not properly cleaned up before `render_vis` returns. This...