train failure in yolov2

Zzh-tju / DIoU-darknet

Distance-IoU Loss into YOLO v3

GNU General Public License v3.0

313 stars 85 forks source link

Open Sylvia-Vee opened 4 years ago

Sylvia-Vee commented 4 years ago

hi 你好请问一下，ciou可以兼容 yolov2吗？我在yolov2上训练的时候w h梯度更新后直接failure掉了

Zzh-tju commented 4 years ago

首先你需要确定是否替换正确，yolov2是region层而v3是yolo层

Sylvia-Vee commented 4 years ago

对，是region层，也不是failure，是收不下去，对lr比较敏感，容易nan

Zzh-tju commented 4 years ago

试试其他loss看看