Is there a Balanced Group Softmax implementation that works with detectron2

[jackkwok]

If not, how can I go about implementing Balanced Group Softmax for detectron2?

[Shanyaodedanshen]

I also want to use BAGS in detectron2.

[twangnh]

hi, it should be easy to port group softmax to detectron2