ReDet icon indicating copy to clipboard operation
ReDet copied to clipboard

单卡训练难以收敛

Open bigFatCatTom opened this issue 3 years ago • 2 comments

我将自己的数据筛选完毕后训练,难以收敛,测试的时候检测不到边框(训练测试都是同比例缩放,观察标注框也没什么问题)。请问下,单卡训练,配置文件需要怎么修改呢。 image

bigFatCatTom avatar Jan 14 '22 02:01 bigFatCatTom

还有个问题就是,作者您给的测试模型ReDet_re50_refpn_1x_dota1大小只有125m,为啥我用同样配置文件训练出来的模型有245m呢,作者后期做了量化么。

bigFatCatTom avatar Jan 14 '22 03:01 bigFatCatTom

  1. All our models are trained on 4GPUs with a learning rate 0.01. If you train your model with more/fewer GPUs, remember to change the learning rate, e.g., 0.02lr=0.0025lr8GPU, 0.0025lr=0.0025lr1GPU.
  1. 我们去除了模型中的optimizer:https://github.com/csuhan/ReDet/blob/master/tools/publish_model.py

csuhan avatar Jan 14 '22 06:01 csuhan