soeaver / caffe-model

Caffe models (including classification, detection and segmentation) and deploy files for famouse networks
MIT License
1.28k stars 623 forks source link

pspnet error: Check failed: status == CUBLAS_STATUS_SUCCESS (11 vs. 0) #65

Open iliadsouti opened 6 years ago

iliadsouti commented 6 years ago

I got an error while running the pspnet_resnet_v2 model with SBD data. Check failed: status == CUBLAS_STATUS_SUCCESS (11 vs. 0) CUBLAS_STATUS_MAPPING_ERROR Can you help me resolve it. Some people mentioned it may be related to the scaling of the label. Since you read the labels with cv2.imread in a grayscale mode in the image_seg_data.py layer. Could it be a result of that?