It's urgent, why is it that my training and reasoning directly using the source code is so slow?

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

1.64k stars 178 forks source link

It's urgent, why is it that my training and reasoning directly using the source code is so slow? #296

Closed txw-github closed 1 month ago

txw-github commented 1 month ago

I'm using the pytorch version on an RTX 3060 and it takes 4-5min for a training dataset of 6,000 sheets and 45s for a test set of 2,000 sheets. whereas yolov8 only takes 1.3min and 17s