Grad-CAM-Pytorch
Grad-CAM-Pytorch copied to clipboard
fix for updated PyTorch (higher than 0.5)
.data[0] -> .data