https://singularityfeed.com/grad-cam-from-scratch-with-pytorch-hooks/