search

Megvii-BaseDetection / OTA

Official implementation of our CVPR2021 paper "OTA: Optimal Transport Assignment for Object Detection" in Pytorch.
Apache License 2.0
242 stars 24 forks source link

Run on single gpu. #10

Open kuazhangxiaoai opened 2 years ago

kuazhangxiaoai commented 2 years ago

Good code! I am revising the code for sing gpu. and ask the of 3 lines of code in losses function(fcos.py).

if self.norm_sync:
    dist.all_reduce(num_foreground)
    num_foreground /= dist.get_world_size()

What's meaning of the 3 lines of code? Can I delete these? Thanks.