Grad-CAM-Pytorch
Grad-CAM-Pytorch copied to clipboard
I met this error when run 6)Train session in GuidedBackprop.ipynb.
Hi, @GunhoChoi
I met this error when run 6)Train session in GuidedBackprop.ipynb.
RuntimeError: Input type (torch.cuda.DoubleTensor) and weight type (torch.cuda.FloatTensor) should be the same
What's wrong with me?
Thanks in advance.
from @bemoregt.
@bemoregt Try to change the input data type into float like the code below. x = Variable(image.float().cuda())
thanks the answer