Added Training Sample code for fasterrcnn_resnet50_fpn (#1695)

torchvision/models/detection/faster_rcnn.py

@@ -318,7 +318,16 @@ def fasterrcnn_resnet50_fpn(pretrained=False, progress=True,
     Example::
 
         >>> model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
-        >>> model.eval()
+        >>> images,boxes,labels = torch.rand(4,3,600,1200), torch.rand(4,11,4), torch.rand(4,11) # For Training
+        >>> images = list(image for image in images)
+        >>> targets = []                                   
+        >>> for i in range(len(images)):
+        >>>     d = {}
+        >>>     d['boxes'] = boxes[i]
+        >>>     d['labels'] = labels[i].type(torch.int64)
+        >>>     targets.append(d)
+        >>> output = model(images,targets)
+        >>> model.eval()                                                                         # For inference
         >>> x = [torch.rand(3, 300, 400), torch.rand(3, 500, 400)]
         >>> predictions = model(x)