yuantn / MI-AOD

Code for Multiple Instance Active Learning for Object Detection, CVPR 2021

Home Page:https://openaccess.thecvf.com/content/CVPR2021/papers/Yuan_Multiple_Instance_Active_Learning_for_Object_Detection_CVPR_2021_paper.pdf

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RuntimeError: SigmoidFocalLoss is not compiled with GPU support

chiran7 opened this issue · comments

Dear Author,

after command : python tools/train.py configs/MIAOD.py, It shows the following error:
File "/home/cshah/.local/lib/python3.8/site-packages/mmcv/ops/focal_loss.py", line 54, in forward
ext_module.sigmoid_focal_loss_forward(
RuntimeError: SigmoidFocalLoss is not compiled with GPU support.

I have installed mmcv and mmdetection. Can you suggest the better solution to resolve this issue ?

Thank you for your time and consideration.

Please refer to Issue #81, and READ IT CAREFULLY.