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

还有个问题就是,作者您给的测试模型ReDet_re50_refpn_1x_dota1大小只有125m,为啥我用同样配置文件训练出来的模型有245m呢,作者后期做了量化么。
- 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.
- 我们去除了模型中的optimizer:https://github.com/csuhan/ReDet/blob/master/tools/publish_model.py