Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

pytorch版本复现，没有达到论文精度 #182

Closed
1747956LXR opened this issue Jan 16, 2024 · 1 comment
Closed

pytorch版本复现，没有达到论文精度 #182

1747956LXR opened this issue Jan 16, 2024 · 1 comment

Comments

@1747956LXR
Copy link

我使用一张3090显卡复现pytorch版的rt-detr，复现的模型为rtdetr_r18vd_6x_coco.yml，设置的batchsize=16，其他参数和代码都没有改变，最后得到45.9的mAP，与论文报告的46.5的mAP还有差距，以下是训练最后几个epoch的截图，有劳作者帮忙看一下，谢谢！
image
@lyuwenyu
Copy link
Owner

lyuwenyu commented Jan 17, 2024 •
edited
Loading

这个版本我只跑了r50的 log放这里了#42

ps.
另外我重新设计了pytorch版本的rtdetrv2_r18的精度mAP 47.9

可以先star这个repo并持续关注一下后续的状态

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@lyuwenyu @1747956LXR