bingykang / Fewshot_Detection

Few-shot Object Detection via Feature Reweighting
https://arxiv.org/abs/1812.01866
526 stars 111 forks source link

Loss becomes nan after few batches #43

Open ashley-mv opened 4 years ago

ashley-mv commented 4 years ago

Selection_320

Hello, The number of proposals are reduced to zeros after a few iterations, and later the loss becomes nan. VOC dataset is used for training. Has anyone run into the same problem?

meijuan1102 commented 4 years ago

Hello, have you solved this problem? I also met the same problem

QJSQJS commented 3 years ago

Hello, I also met the problem+1, :( . Have you solved this problem?

QJSQJS commented 3 years ago

I found a solution! When the batch decreases, the lr should decrease. The following are my parameters: batch = 8; subdivisions = 4; lr = 0.0001;

It works fine!