DIoU-pytorch-detectron icon indicating copy to clipboard operation
DIoU-pytorch-detectron copied to clipboard

Published 20 hours ago • verified publisher icon Zzh-tju

I cannot get similar performance

Open yfzhang114 opened this issue 3 years ago • 2 comments 

I have a question for DIOU code. I use the same code as your repo https://github.com/Zzh-tju/DIoU-pytorch-detectron, while I cannot get the same result. Surprisingly, even when I use the IOU LOSS, I cannot get the same performance as the original article. Have I missed some details? Thank you.

yfzhang114 avatar Sep 20 '20 03:09 yfzhang114

How many GPUs did you use? I used 1 GPU for training coco to 720K iterations. If you use 8 GPUs to 90k iters, it will drop to 36.8 AP I guess.

Zzh-tju avatar Sep 20 '20 11:09 Zzh-tju

I use 4 GPU for training coco to 90k iters, and it drops to 34.7AP without modifying any configure

yfzhang114 avatar Sep 20 '20 12:09 yfzhang114