Update faster_rcnn.py (#979)

Add 'import torch' in example

Update faster_rcnn.py (#979)
Add 'import torch' in example
3d561039 · Gu-ni-kim · Francisco Massa · 47db9638 · 3d561039
Commit 3d561039 authored Jun 03, 2019 by Gu-ni-kim Committed by Francisco Massa Jun 03, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

torchvision/models/detection/faster_rcnn.py torchvision/models/detection/faster_rcnn.py +1 -0

No files found.
--- 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