YOLOv5-6.1训练一个四分类模型时出现如下错误,请问大家如何修改
File "F:\yolo\road_detect\yolov5-v6.1\utils\loss.py", line 148, in __call__
t[range(n), tcls[i]] = self.cp
IndexError: index 9 is out of bounds for dimension 1 with size 4
YOLOv5-6.1训练一个四分类模型时出现如下错误,请问大家如何修改
File "F:\yolo\road_detect\yolov5-v6.1\utils\loss.py", line 148, in __call__
t[range(n), tcls[i]] = self.cp
IndexError: index 9 is out of bounds for dimension 1 with size 4
分类请运行yolov5/classify/train.py这个文件,而不是yolov5/train.py,这两个是不一样的