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.31k stars 258 forks source link

训练轮数问题 #410

Open 20231211 opened 1 month ago

20231211 commented 1 month ago

设置随机种子后每次跑出的结果还有近2个点的差距，是因为训练epoch不够吗？配置文件默认是跑72轮是因为加载预训练权重吗？如果不加载预训练权重建议训练多少个epoch？

fym202 commented 1 month ago

你的这个差距有点大呀，我在小数据集上跑的时候确实需要加多epoch才能更好，v1的R18模型从72个epoch提升到100epoch提升了0.8的AP。我现在是正在复现v2的R18模型在coco上的结果。

20231211 commented 1 month ago

@fym202 你好，请问你的数据集大概有多少张图片？

fym202 commented 1 month ago

我就是跑的coco数据集，然后现在主要是在coco25k数据集（从coco里采样了25k张数据）上跑的。

20231211 commented 1 month ago

谢谢！

---原始邮件--- 发件人: @.> 发送时间: 2024年8月16日(周五) 晚上9:18 收件人: @.>; 抄送: @.**@.>; 主题: Re: [lyuwenyu/RT-DETR] 训练轮数问题 (Issue #410)

我就是跑的coco数据集，然后现在主要是在coco25k数据集（从coco里采样了25k张数据）上跑的。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>