This code is originally from: 
https://github.com/jacobgil/pytorch-grad-cam

