search

kazuto1011 / grad-cam-pytorch

PyTorch re-implementation of Grad-CAM (+ vanilla/guided backpropagation, deconvnet, and occlusion sensitivity maps)
MIT License
784 stars 174 forks source link

Change the hook func in Grad-CAM #14

Closed kazuto1011 closed 5 years ago

kazuto1011 commented 5 years ago

To save feature maps and gradients of the specific layer, now use the module's name instead of the object identity.