YOLO7 indices should be either on cpu or on the same device as the indexed tensor (cpu)

在loss.py文件解决办法: 在loss.py文件中增加下图中第一行,修改下面二三行

device = targets.device
在这里插入图片描述
from_which_layer.append((torch.ones(size=(len(b),)) * i).to(torch.device(device)))
在这里插入图片描述
fg_mask_inboxes = fg_mask_inboxes.to(torch.device(device))
在这里插入图片描述

你可能感兴趣的:(python,开发语言)