From b2b1e04ee141bfb02e3ebf7a2d4533f9ecc29979 Mon Sep 17 00:00:00 2001
From: Jackson Liu <47380172+JacksonLiu-ai@users.noreply.github.com>
Date: Tue, 28 Apr 2020 11:12:21 +0800
Subject: [PATCH] Update rpn.py

---
 torchvision/models/detection/rpn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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).