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.31k stars 258 forks source link

关于训练时长的问题 #278

Open lkxxxxx opened 5 months ago

lkxxxxx commented 5 months ago

我自己的数据集共2700多张，我在3090和4090上都训练过，bs都是16，imgsz是640，在训练的时候大概15分钟一个epoch,有没有什么方法可以加快训练速度，多谢解答！！！

lyuwenyu commented 5 months ago

增大bs，多卡训练，混合精度训练，减小imgsz，等等

txw-github commented 4 months ago

您好，你的FPS正常吗

txw-github commented 4 months ago

为什么我在RTX3060上使用pytorch版本，训练数据集6000张需要4-5min，测试集2000张需要45s。而yolov8只要1.3min和17s，请求大佬指导下，比较急需

lkxxxxx commented 4 months ago

@txw-github 你的bs设置的多大，图片resize是多少，这些都有关系，yolov5和v8我都试过，同样的数据集和设置v8比v5确实训练速度都要快不少，我觉得应该是模型本身有一部分原因，代码的优化也有一定的原因。FPS我使用的3090跑的RT-DETR-r18的版本，使用的onnx模型，没有trt加速。大概是40多点。