tztztztztz / eql.detectron2

The official implementation of Equalization Loss for Long-Tailed Object Recognition (CVPR 2020) based on Detectron2. https://arxiv.org/abs/2003.05176
Apache License 2.0
202 stars 17 forks source link

How to reproduce results on R-50-C4 and R-101-C4 in your paper? #10

Open U-Help opened 3 years ago

U-Help commented 3 years ago

Well, in your paper, there are results on different frameworks and models, including R-50-C4 and R-101-C4. But in this repo, you don't provide relevant results. I try to implement this by writing a file named Base-EQL-RCNN-C4.yaml, however, the training process always ends with the following bugs:

FloatingPointError: Loss became infinite or NaN at iteration=2! loss_dict = {'loss_cls': tensor(nan, device='cuda:0', grad_fn=), 'loss_box_reg': tensor(nan, device='cuda:0', grad_fn=), 'loss_mask': tensor(0.6931, device='cuda:0', grad_fn=), 'loss_rpn_cls': tensor(nan, device='cuda:0', grad_fn=), 'loss_rpn_loc': tensor(inf, device='cuda:0', grad_fn=)}

Can you give me some help? Base-EQL-RCNN-C4.zip