search

jbwang1997 / OBBDetection

OBBDetection is an oriented object detection library, which is based on MMdetection.
Apache License 2.0
522 stars 112 forks source link

train-loss = Nan #100

Open xiyufeng2 opened 2 years ago

xiyufeng2 commented 2 years ago

2021-12-20 17:13:30,133 - mmdet - INFO - Epoch [12][150/4867] lr: 5.000e-05, eta: 0:29:00, time: 0.400, data_time: 0.005, memory: 4601, loss_rpn_cls: nan, loss_rpn_bbox: nan, loss_cls: nan, acc: 11.5102, loss_bbox: nan, loss: nan, grad_norm: nan all loss is always nan.what's wrong?

jbwang1997 commented 2 years ago

Hello, can you provide more information?

xiyufeng2 commented 2 years ago

I didn't load pretrained weight to train the model , leading the loss to be large. I think the weight init method is not suitable