OBBDetection icon indicating copy to clipboard operation
OBBDetection copied to clipboard

train-loss = Nan

Open xiyufeng2 opened this issue 2 years ago • 2 comments

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?

xiyufeng2 avatar Dec 20 '21 09:12 xiyufeng2

Hello, can you provide more information?

jbwang1997 avatar Dec 21 '21 07:12 jbwang1997

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

xiyufeng2 avatar Dec 28 '21 03:12 xiyufeng2