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

batch_size 和 num_workers的问题 #290

Open DaCheng1823 opened 5 months ago

DaCheng1823 commented 5 months ago

我想知道

coco_detection.yml、dataloader.yml

里面的 batch_size 和 num_workers的分别的作用，两个设置的内容可以不一样吗？

（已经点击star）

lyuwenyu commented 5 months ago

dataloader.yml 会覆盖 coco_detection.yml 里的内容，只需要在dataloader.yml更改即可

( ps https://github.com/lyuwenyu/RT-DETR/issues/287 这个问题是让你按照模板要求描述bug

SoraJung commented 3 months ago

I modified batch size, num worker in only coco_detection.yml. (pytorch) Should I change batch size, num worker in only dataloader.yml?