yjh0410 / PyTorch_YOLOv3

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

the loss is nan

bbbeach opened this issue · comments

I train the yolov3 model with voc2007 and voc 2012.In the second epoch,the loss turn up nan

Could you tell what your batch size is?

my batch size is 16, using the original learning rate with warmup strategy

Ok, I see. I will check the nan problem. Before that, you can try my another YOLO project: https://github.com/yjh0410/yolov2-yolov3_PyTorch. @bbbeach

thanks