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