YOLOv7 训练报错 RuntimeError: result type Float can‘t be cast to the desired output type __int64

遇到这种问题不要慌，按我的步骤试一下

第一步：

在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()

问题解决，开始训练你的数据集吧