使用YOLOV5训练自己的数据集过程中遇到的问题

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() ”即可。