logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

一、 Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend解决方法.

方法一:上述链接中内容比较详细,我这里简要提一下,在第一步框选的路径下,创建新的文件夹,内容放在文章末尾,然后命名xx(随意),然后在上图return返回值部分,改为 return xx.nms就行了(但是我自己的这样改,还是报错,没有找到声明,我不知道如何解决qaq,浪费了n小时时间,所以有了第二个方案)lt = torch.max(boxes1[:, None, :2], boxes2[:,

文章图片
#python#pytorch
到底了