l0ngjx

If I want to set different conf-thre for different tags, what should I do? 如果我想要给不同标签设置不同conf-thre，我应该怎么做呢？ auto conf_mask = detections.select(2, 4).ge(conf_thres).unsqueeze(2); or cv::dnn::NMSBoxes(offset_box_vec, score_vec, conf_thres, iou_thres, nms_indices);？

我在运行模型时第50行出现了cuda错误，我的torch版本是1.6.0，但我不确定模型需要哪一个numpy版本？