diff --git a/torchvision/models/detection/faster_rcnn.py b/torchvision/models/detection/faster_rcnn.py index d38c870e26d..057fb2bf221 100644 --- a/torchvision/models/detection/faster_rcnn.py +++ b/torchvision/models/detection/faster_rcnn.py @@ -99,6 +99,7 @@ class FasterRCNN(GeneralizedRCNN): Example:: + >>> import torch >>> import torchvision >>> from torchvision.models.detection import FasterRCNN >>> from torchvision.models.detection.rpn import AnchorGenerator