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 259 forks source link

单卡RTX4090训练时间问题 #226

Open wyk777777 opened 6 months ago

wyk777777 commented 6 months ago
Snipaste_2024-03-08_18-06-35

这个训练速度是正常的嘛？还是我没有成功调用GPU？使用了#train on single-gpu中提到的脚步运行

lyuwenyu commented 6 months ago

单卡的话batch_size * 4，给的配置是4卡跑的

AntAdventure commented 6 months ago

@wyk777777 你好，我想了解一些你单卡4090训练模型要花费多长时间

wyk777777 commented 6 months ago

@AntAdventure 我没有训练完因为目测需要的时间太久了 我使用的是3090（因为是学校的算力大概只有3080的性能）一轮可能就需要2天（原来我前面写成了4090没有注意到）

chhchhchh commented 6 months ago

@lyuwenyu 作者训练coco数据集大概用了多大的显存资源

txw-github commented 4 months ago

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

Zhonghui-Gao commented 3 weeks ago

Snipaste_2024-03-08_18-06-35 这个训练速度是正常的嘛？还是我没有成功调用GPU？使用了#train on single-gpu中提到的脚步运行

我的也是这样在windows上感觉没用到GPU 因为训练真的很慢

wyk777777 commented 3 weeks ago

应该是用到gpu了吧 不用gpu速度会比这慢得多。可能yolov8就是比较快。具体我也不是很清楚因为我研究方向也不是图像

---- 回复的原邮件 ---- | 发件人 | @.> | | 发送日期 | 2024年09月06日 12:33 | | 收件人 | lyuwenyu/RT-DETR @.> | | 抄送人 | wyk777777 @.>, Mention @.> | | 主题 | Re: [lyuwenyu/RT-DETR] 单卡RTX4090训练时间问题 (Issue #226) |

这个训练速度是正常的嘛？还是我没有成功调用GPU？使用了#train on single-gpu中提到的脚步运行

我的也是这样在windows上感觉没用到GPU 因为训练真的很慢

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>