Expected all tensors to be on the same device,but found at least two device,cuda:0 and cpu

2020.12.16
遇到问题

想把检测时候的二次筛选只进行一次。所以第一次不在检测区域里的检测框不进行删除。

把之前的加了检测区域检测的代码删除后：

报错：

tensor不在同一个设备上，在scf后面加上.to(self.device),又报了新的错误：

这个是因为把CUDA tensor格式的数据转为numpy时，需要先将其转为cpu float-tensor然后再转为numpy格式。

把前面的scf改回来，然后if条件句中改为以下。因为发现input_size和image_size都是1024，相减为0，所有就将前两行代码注释了。

运行成功。

是否还有更好的解决办法？

2021.12.20
有，针对tensor不在同一个设备上，在scf后面加上.to(self.device),会报了新的错误，scf不变，将detected转为cpu，程序可正常运行。