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

Inception_v3 error #2

Closed zhangrenyu closed 6 years ago

zhangrenyu commented 6 years ago

The inception_v3 does not has a relu layer, but it has a F.relu function in its forward pass. You cannot update the relu in this way for inception_v3.

kazuto1011 commented 6 years ago

Thank you for reporting the bug. I have missed the point. In the results directory, you can see the "Guided Backpropagation" examples through the Inception model are similar to vanilla ones. I will update readme sometime soon, instead of changing the updating manner.