lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

有关decoder的消融实验

lkxxxxx opened this issue · comments

commented

论文中decoder的消融实验介绍模型可以无需重新训练而使用不同的解码器层,请问一下我需要调整哪里的参数来更换decoder层数?
我在 num_decoder_layers直接更换decoder层数时验证模型会报错。
image
image

不用改结构,推理只需要调整eval_idx就行

commented

@lyuwenyu 好的,非常感谢。