1.使用YOLOV5训练自己的数据集时,运行train.py是报错:RuntimeError: result type Float can‘t be cast to the desired output type __int64
解决办法:将utils文件夹下的loss.py中gain = torch.ones(7, device=targets.device)改为gain = torch.ones(7, device=targets.device).long()即可。原因是新版本的torch无法自动执行此转换,旧版本torch可以。
2.使用YOLOV5训练自己的数据集时,运行train.py是报错:TypeError:'builtin_function_or_method' object does not support item assignment
解决办法:将loss.py中的“gain = torch.ones(7, device=targets.device).long”改为“gain = torch.ones(7, device=targets.device).long() ”即可。