文章目录
在使用yolov5训练自己的数据集时,运行train.py时出现上述问题
解决方式:
将loss.py中gain = torch.ones(7, device=targets.device)
改为gain = torch.ones(7, device=targets.device).long()
即可。原因是新版本的torch无法自动执行此转换,旧版本torch可以。
loss.py在utils文件夹下,ctrl+f搜索gain,找到gain = torch.ones(7, device=targets.device),将其修改为gain = torch.ones(7, device=targets.device).long(),问题解决