Fix argument name in doctstring (#2093)

9bb43c62 · theotheo · GitHub · 4f162503 · 9bb43c62
Unverified Commit 9bb43c62 authored Apr 14, 2020 by theotheo Committed by GitHub Apr 14, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- a/torchvision/models/detection/generalized_rcnn.py
+++ b/torchvision/models/detection/generalized_rcnn.py
@@ -18,7 +18,7 @@ class GeneralizedRCNN(nn.Module):
    Arguments:
        backbone (nn.Module):
        rpn (nn.Module):
-        heads (nn.Module): takes the features + the proposals from the RPN and computes
+        roi_heads (nn.Module): takes the features + the proposals from the RPN and computes
            detections / masks from it.
        transform (nn.Module): performs the data transformation from the inputs to feed into
            the model