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

打印模型结构 #250

Open lei12879 opened 7 months ago

lei12879 commented 7 months ago

你好，我想问一下rtdetr的pytorch版本在训练开始是没有打印模型结构吗，如果是的话，我想问一下我想看模型结构，代码应该如何改呢

lyuwenyu commented 7 months ago

直接用 print()

lei12879 commented 7 months ago

具体我应该加到那个位置呢