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

单卡3090ti，源码pytorch版本训练问题 #301

Open Acc2386 opened 4 months ago

Acc2386 commented 4 months ago

已star，您好，请问下：源码pytorch版本，按照您的步骤设置好了。train如何提高bath_size和num_workers，显存和gpu利用率都不变，并且发现cpu在训练，爆内存是设置太高了的问题。似乎没用gpu，我改如何处理这个问题呢？

lyuwenyu commented 4 months ago

看下你gpu是不是正确使用了 https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/src/core/config.py#L75

Acc2386 commented 4 months ago

看下你gpu是不是正确使用了 https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/src/core/config.py#L75

你好，谢谢，今天重装环境跑通了