search

chensnathan / YOLOF

You Only Look One-level Feature (YOLOF), CVPR2021, Detectron2
MIT License
271 stars 28 forks source link

a map #16

Closed wanghangege closed 3 years ago

wanghangege commented 3 years ago

I used the base training file (Base-YOLOF.yaml) and obtained a training accuracy of about 14%. Using the best training file (YOLOF_CSP_D_53_DC5_9x_stage2_3x.yaml) in the table, I got about 38% mAP. I used two titanV graphics cards, so adjusted the batch, not sure if that affected it.

chensnathan commented 3 years ago

The settings are made for 8 GPUs, when you have only 2 GPUs, you should adjust the learning rate, the iterations, the warmup iterations, and the warmup learning rate factor according to the linear scaling rule.

wanghangege commented 3 years ago

The settings are made for 8 GPUs, when you have only 2 GPUs, you should adjust the learning rate, the iterations, the warmup iterations, and the warmup learning rate factor according to the linear scaling rule.

thanks, I forgot it.