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

预训练权重 #365

Closed lime-s closed 4 months ago

lime-s commented 4 months ago

您好，我想问一下，我使用下述命令进行训练：

training on single-gpu

export CUDA_VISIBLE_DEVICES=0 python tools/train.py -c configs/rtdetr/rtdetr_r50vd_6x_coco.yml 我没有指定预训练权重，这个下载的预训练权重是在哪里定义的呢

lyuwenyu commented 4 months ago

https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/src/nn/backbone/presnet.py#L196