如果sigmoid_focal_loss_forward_impl 是一个自定义的CUDA算子,确保你已经正确编译了这个算子,并且编译后的库文件(如 .so 或.dll)已经正确加载。 检查编译设置,确保它们与你的PyTorch和CUDA版本相匹配。 查看文档和源代码: 如果这个函数是某个库的一部分,查阅该库的文档或源代码,确认是否有针对CUDA设备的实现。
Whitespace Ignore whitespace Split Unified 188 changes: 0 additions & 188 deletions 188 maskrcnn_benchmark/csrc/SigmoidFocalLoss_cuda.cu Load diff This file was deleted. 0 comments on commit 4d1ba33 Please sign in to comment. Footer...