Skip to content

Commit cae5400

Browse files
authored
Replace functional.sigmoid with torch.sigmoid (#3307)
Removes deprecation warning
1 parent d732562 commit cae5400

File tree

1 file changed

+1
-1
lines changed
  • torchvision/models/detection

1 file changed

+1
-1
lines changed

torchvision/models/detection/rpn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -252,7 +252,7 @@ def filter_proposals(self, proposals, objectness, image_shapes, num_anchors_per_
252252
levels = levels[batch_idx, top_n_idx]
253253
proposals = proposals[batch_idx, top_n_idx]
254254

255-
objectness_prob = F.sigmoid(objectness)
255+
objectness_prob = torch.sigmoid(objectness)
256256

257257
final_boxes = []
258258
final_scores = []

0 commit comments

Comments
 (0)