pytorch_solov2 icon indicating copy to clipboard operation
pytorch_solov2 copied to clipboard

请问有没有遇到这个问题 ImportError: cannot import name 'sigmoid_focal_loss_cuda'

Open qumj opened this issue 3 years ago • 1 comments

Traceback (most recent call last): File "G:/pycharmProject/pytorch_solov2-master/train.py", line 4, in from modules.solov2 import SOLOV2 File "G:\pycharmProject\pytorch_solov2-master\modules\solov2.py", line 6, in from .solov2_head import SOLOv2Head File "G:\pycharmProject\pytorch_solov2-master\modules\solov2_head.py", line 8, in from .focal_loss import FocalLoss File "G:\pycharmProject\pytorch_solov2-master\modules\focal_loss.py", line 3, in from modules.sigmoid_focal_loss import sigmoid_focal_loss as sigmoid_focal_loss File "G:\pycharmProject\pytorch_solov2-master\modules\sigmoid_focal_loss_init.py", line 1, in from .sigmoid_focal_loss import SigmoidFocalLoss, sigmoid_focal_loss File "G:\pycharmProject\pytorch_solov2-master\modules\sigmoid_focal_loss\sigmoid_focal_loss.py", line 5, in from . import sigmoid_focal_loss_cuda ImportError: cannot import name 'sigmoid_focal_loss_cuda'

qumj avatar Aug 04 '22 09:08 qumj

这是windows下编译的嘛 我没有在windows下验证过,那个编译sigmoid_focal_loss_cuda可能只在Linxu下可以成功

ABlueLight avatar Sep 21 '22 02:09 ABlueLight