search

DevashishPrasad / CascadeTabNet

This repository contains the code and implementation details of the CascadeTabNet paper "CascadeTabNet: An approach for end to end table detection and structure recognition from image-based documents"
MIT License
1.46k stars 422 forks source link

cuda runtime error (209) : unrecognized error code at mmdet/ops/roi_align/src/roi_align_kernel.cu:139 #88

Open tiwari93 opened 3 years ago

tiwari93 commented 3 years ago

While running the inference getting error :

RuntimeError Traceback (most recent call last)

in () 12 13 # Run Inference ---> 14 result = inference_detector(model, img) 15 16 # Visualization results 11 frames /content/drive/My Drive/repos/mmdetection/mmdet/ops/roi_align/roi_align.py in forward(ctx, features, rois, out_size, spatial_scale, sample_num, aligned) 34 out_w) 35 roi_align_cuda.forward_v1(features, rois, out_h, out_w, ---> 36 spatial_scale, sample_num, output) 37 else: 38 output = roi_align_cuda.forward_v2(features, rois, RuntimeError: cuda runtime error (209) : unrecognized error code at mmdet/ops/roi_align/src/roi_align_kernel.cu:139 **Running this on google colab**