Grad-CAM-Pytorch
Grad-CAM-Pytorch copied to clipboard
fix data type error
fix code to solve the error Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same