遇到这种问题不要慌,按我的步骤试一下
第一步:
在utils文件夹下找到loss.py或者yolo_training.py等,官方文件是loss.py,不同博主复现代码,可能其不同名字。
第二步:
Ctrl+f搜索gain,找到gain = torch.ones(7, device=targets.device)这段代码。
第三步:
将其修改为gain = torch.ones(7, device=targets.device).long()
就是代码最后加.long()
问题解决,开始训练你的数据集吧