Replace functional.sigmoid with torch.sigmoid (#3307)

Removes deprecation warning

Replace functional.sigmoid with torch.sigmoid (#3307)
Removes deprecation warning
cae5400d · Francisco Massa · GitHub · d732562f · cae5400d
Unverified Commit cae5400d authored Jan 27, 2021 by Francisco Massa Committed by GitHub Jan 27, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/models/detection/rpn.py torchvision/models/detection/rpn.py +1 -1

No files found.
--- a/torchvision/models/detection/rpn.py
+++ b/torchvision/models/detection/rpn.py
@@ -252,7 +252,7 @@ class RegionProposalNetwork(torch.nn.Module):
        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 = []