search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.61k stars 303 forks source link

模型微调 #244

Open lebron-2016 opened 7 months ago

lebron-2016 commented 7 months ago

作者您好，请问rtdetr_pytorch配置文件中对学习率的设置是与GPU数量相关的吗？如果要在自己的数据集上进行微调，应该如何设置学习率？感谢！！

lyuwenyu commented 7 months ago

建议训练先不要改learning_rate, 如果不符合预期再调。( ps. 部分经验是在任何数据上以及是不是微调 目前配置都是最好的，this is a feature...

lebron-2016 commented 7 months ago

建议训练先不要改learning_rate, 如果不符合预期再调。( ps. 部分经验是在任何数据上以及是不是微调 目前配置都是最好的，this is a feature...

好的，谢谢您！！

我目前微调时遇到的情况是，验证集精确率在逐epoch下降（第一个epoch精确率是最高的），召回率在逐epoch上升，这种情况应该如何处理？

感谢！！