ternaus/TernausNetV2

runtime error: CUDNN_STATUS_MAPPING_ERROR

HIWFSS opened this issue · 1 comments

HIWFSS commented

When I execute “prediction = F.sigmoid(model(input_img)).data[0].cpu().numpy()” is reporting error “runtime error: CUDNN_STATUS_MAPPING_ERROR”，what should I do ? thank you very much!

saranshkarira commented

its using PyTorch 0.4, try instead:
prediction = torch.sigmoid(model(input_img)).item()