avBuffer / Yolov5_tf

Yolov5/Yolov4/ Yolov3/ Yolo_tiny in tensorflow
290 stars 108 forks source link

train yolov5 get nan #15

Open TaiwanRobert opened 3 years ago

TaiwanRobert commented 3 years ago

Hi man, I had train yolov5. now had a problem ,the loss always random getting nan. maybe the problem from iou???

payal211 commented 3 years ago

Hi @TaiwanRobert I have same issue. Can you please give details of your configuration and system environment? please explain, How you able to run train.py with net_type: yolov5 and YOLO.NET_TYPE: "darknet53"?

Thanks.