diff --git a/torchvision/models/detection/rpn.py b/torchvision/models/detection/rpn.py index 9f20b235863..3a9b014db5d 100644 --- a/torchvision/models/detection/rpn.py +++ b/torchvision/models/detection/rpn.py @@ -458,7 +458,7 @@ def forward(self, images, features, targets=None): """ Arguments: images (ImageList): images for which we want to compute the predictions - features (List[Tensor]): features computed from the images that are + features (OrderedDict[Tensor]): features computed from the images that are used for computing the predictions. Each tensor in the list correspond to different feature levels targets (List[Dict[Tensor]]): ground-truth boxes present in the image (optional).