Update comments regarding difference among object detection models

pytorch · Dec 18, 2024 · 0595f76 · 0595f76
1 parent f7b1cfa
commit 0595f76
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/source/models.rst b/docs/source/models.rst
@@ -420,6 +420,7 @@ Here is an example of how to use the pre-trained object detection models:
 
     # Step 1: Initialize model with the best available weights
     weights = FasterRCNN_ResNet50_FPN_V2_Weights.DEFAULT
+    # for FCOS, RetinaNet, SSD and SSDlite, use `score_thresh=0.9`. 
     model = fasterrcnn_resnet50_fpn_v2(weights=weights, box_score_thresh=0.9)
     model.eval()