Some problems in training

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

GNU General Public License v3.0

13.31k stars 4.2k forks source link

Some problems in training #223

Open M15-3080 opened 2 years ago

M15-3080 commented 2 years ago

May I ask what is going on? What causes it? This did not happen with YOLOv5

WongKinYiu commented 2 years ago

I have not meet this problem. I think it may due to we do not check label status in simota or divide zero bugs in ciou loss. Temporally you could stop training and resume training with your_exp/weights/best.pt to avoid the problem.

By the way, for finetuning model using cfg/training/yolov7-w6.yaml, it is better to use yolov7-w6_training.pt.