hyz-xmaster / VarifocalNet

VarifocalNet: An IoU-aware Dense Object Detector
Apache License 2.0
351 stars 52 forks source link

cls loss is increasing #32

Closed jacksonsc007 closed 2 years ago

jacksonsc007 commented 2 years ago

Hi, @hyz-xmaster I noticed that the cls_loss using varifocal loss does not decrease until roughly 1000 iterations. Could you give us some insights?

hyz-xmaster commented 2 years ago

You may try a lower learning rate or more powerful features like using DCN to extracting features to predict IoU-aware cls scores.

jacksonsc007 commented 2 years ago

@hyz-xmaster Thanks for your prompt reply. I checked your training log and found that the cls_loss kept loitering around for 700 iterations, is this normal or a problem we need to address?

hyz-xmaster commented 2 years ago

@jacksonsc007 Well, there is a warm-up stage in training where the learning rate increases linearly from a very small value to the initial learning rate. In general, this process takes 500 iterations. See here . During this stage, the losses move up and down. So it's normal.

jacksonsc007 commented 2 years ago

@hyz-xmaster Thank you.