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

将epoch的值引入RT-DETR的head中 #426

Open
sdreamforchen opened this issue Aug 22, 2024 · 1 comment
Open

将epoch的值引入RT-DETR的head中 #426

sdreamforchen opened this issue Aug 22, 2024 · 1 comment
Assignees
@lyuwenyu

Comments

@sdreamforchen
Copy link

您好，想请教一下，如何将训练的epoch/iteration配置到RT-DETR的head中呢？ 类似output=αA+（1-α）B，α=epoch/total_epoch；因为现有的architectures架构是将backbone、neck和head一起传到train里面的，不是很懂如何单独对head进行操作，且能传train中的epoch/iteration参数；能否给一个示例，谢谢！
@lyuwenyu
Copy link
Owner

把head的forward改造成可以接收epoch的参数；同理 改造模型forward；然后在训练过程中 传入就行吧

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@lyuwenyu lyuwenyu

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@lyuwenyu @sdreamforchen