d0coco

RuntimeError: weight tensor should be defined either for all or no classes

18

I have two categories and I encountered this error when starting training, can you help me Traceback (most recent call last): File "D:/git/python/yolov5ds-main/trainds.py", line 372, in train segloss = SegLoss(seg_pred[1],...