diff --git a/torchvision/models/detection/rpn.py b/torchvision/models/detection/rpn.py index 9ea05c94136..736c82a9009 100644 --- a/torchvision/models/detection/rpn.py +++ b/torchvision/models/detection/rpn.py @@ -252,7 +252,7 @@ def filter_proposals(self, proposals, objectness, image_shapes, num_anchors_per_ levels = levels[batch_idx, top_n_idx] proposals = proposals[batch_idx, top_n_idx] - objectness_prob = F.sigmoid(objectness) + objectness_prob = torch.sigmoid(objectness) final_boxes = [] final_scores = []