search

leftthomas / GradCAM

A PyTorch implementation of Grad-CAM based on ICCV 2017 paper "Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization"
56 stars 13 forks source link

Doesn't work for different models. #2

Closed veritas9872 closed 5 years ago

veritas9872 commented 5 years ago

I tried the program on the resnet 34 architecture with 100 classes and it simply did not work. Apparently, the matrix transpose operation could not be performed. The size mismatch was 512x1 to 512x100.

leftthomas commented 5 years ago

@veritas9872 you should change it by yourself