The confidence loss (obj_loss) scales of the training and validation sets are inconsistent. #13012
Open
1 task done
Labels
question
Further information is requested
Search before asking
Question
Hi, recently I discovered an issue when using the YOLOv5 S model to train my dataset. Although the confidence loss (obj_loss) of the model has decreased in both the training and validation sets, the validation set loss is approximately twice that of the training set loss. However, the bounding box regression loss (box_loss) does not have such an issue. I carefully checked the code and confirmed that the calculation of the loss was correct. May I ask what is the reason for the different scales? Looking forward to your answer.
Additional
No response
The text was updated successfully, but these errors were encountered: