From cc92c0b6a7d057bb52126de9094038fdb81dc6eb Mon Sep 17 00:00:00 2001
From: Gu-ni-kim <dldldlfma@gmail.com>
Date: Sun, 2 Jun 2019 18:52:39 +0900
Subject: [PATCH] Update faster_rcnn.py

Add 'import torch' in example
---
 torchvision/models/detection/faster_rcnn.py | 1 +
 1 file changed, 1 insertion(+)

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