TernausNetV2 icon indicating copy to clipboard operation
TernausNetV2 copied to clipboard

Published 20 hours ago • verified publisher icon ternaus

runtime error: CUDNN_STATUS_MAPPING_ERROR

Open HIWFSS opened this issue 7 years ago • 1 comments

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!

HIWFSS avatar Aug 24 '18 12:08 HIWFSS

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

saranshkarira avatar Sep 18 '18 21:09 saranshkarira