Bobo-y / flexible-yolov5

More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam,dcn and so on), and tensorrt

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cls_loss is always 0

a227799770055 opened this issue · comments

Hi, I training my own dataset with three classes, but the cls_loss is 0 all the time.
Can you help me to figure out the problem?
Thx!

image
image

commented

@a227799770055 i need more info. The nc in model_xxx.yaml is already set 3?

@Bobo-y
Thanks a lot, I forget to set model_xxx.yaml.
Now the training is normal.