解决办法参照以下链接,亲测有效
替换 yolo7/utils/loss.py文件中
"from_which_layer.append((torch.ones(size=(len(b),)) * i)"
为 "from_which_layer.append((torch.ones(size=(len(b),)) * i).to('cuda'))",
在每行 "fg_mask_inboxes = matching_matrix.sum(0) > 0.0" 代码后面添加一行新的代码"fg_mask_inboxes = fg_mask_inboxes.to(torch.device('cuda'))"
示例如下:
fg_mask_inboxes = matching_matrix.sum(0) > 0.0
fg_mask_inboxes = fg_mask_inboxes.to(torch.device('cuda'))
有多处需要添加