search

shaoshengsong / MobileNetV3-SSD

MobileNetV3-SSD for object detection and implementation in PyTorch
269 stars 88 forks source link

Loss inf when traning #4

Open phucnhs opened 5 years ago

phucnhs commented 5 years ago

I'm having a problem when training ssd mobilenet v3 lite, I have lost inf at some steps. Does that affect my results? ---2019-08-22 13:54:37,701 - root - INFO - Epoch: 6, Step: 20, Average Loss: inf, Average Regression Loss inf, Average Classification Loss: 3.2975 2019-08-22 13:56:20,223 - root - INFO - Epoch: 6, Step: 30, Average Loss: inf, Average Regression Loss inf, Average Classification Loss: 3.3092 2019-08-22 13:58:55,497 - root - INFO - Epoch: 6, Step: 40, Average Loss: 6.5821, Average Regression Loss 3.2743, Average Classification Loss: 3.3079 2019-08-22 14:00:47,632 - root - INFO - Epoch: 6, Step: 50, Average Loss: 6.5586, Average Regression Loss 3.2818, Average Classification Loss: 3.2768 +I would like to ask if I download pretrain model voc but when I tested it, I did not return the results of the boxes I thank

shaoshengsong commented 4 years ago

It is recommended that git clone this version https://github.com/shaoshengsong/MobileNetV3-SSD-Compact-Version