It uses 3  anchors_per_location , but I think you’re right. It is a very similar implementation. However, while PyTorch’s implementation returns a logit tensor in the format  [batch, anchors, height, width] , the one I am using returns a logit tensor in the format  [batch, anchors per location * 2, height, width] . Any ideas why the double number of anchor logits? I’m still analyzing the code and I’ll give a feedback as soon as I have a (maybe partial) conclusion
Thanks