问题描述
Traceback (most recent call last):
File “D:\python\modelsmy\saliencymap.py”, line 24, in compute_jacobian
output.backward(mask, retain_graph=True)
RuntimeError: invalid gradient at index 0 - expected type
TensorOptions(dtype=float, device=cuda:0, layout=Strided, requires_grad=false (default), pinned_memory=false (default), memory_format=(nullopt))
but got
TensorOptions(dtype=float, device=cpu, layout=Strided, requires_grad=false (default), pinned_memory=false (default), memory_format=(nullopt))
虚惊一场 只是设备报错
原代码
output.backward(mask, retain_graph=True)
修改后
output.backward(torch.tensor(mask).cuda(), retain_graph=True)