Update docstring example for resnet_fpn_backbone (#7957)

75046bde · Sergii Dymchenko · GitHub · 0b41ff0b · 75046bde
Unverified Commit 75046bde authored Sep 14, 2023 by Sergii Dymchenko Committed by GitHub Sep 14, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

torchvision/models/detection/backbone_utils.py torchvision/models/detection/backbone_utils.py +3 -1

No files found.
--- a/torchvision/models/detection/backbone_utils.py
+++ b/torchvision/models/detection/backbone_utils.py
@@ -79,8 +79,10 @@ def resnet_fpn_backbone(

    Examples::

+        >>> import torch
+        >>> from torchvision.models import ResNet50_Weights
        >>> from torchvision.models.detection.backbone_utils import resnet_fpn_backbone
-        >>> backbone = resnet_fpn_backbone('resnet50', weights=ResNet50_Weights.DEFAULT, trainable_layers=3)
+        >>> backbone = resnet_fpn_backbone(backbone_name='resnet50', weights=ResNet50_Weights.DEFAULT, trainable_layers=3)
        >>> # get some dummy image
        >>> x = torch.rand(1,3,64,64)
        >>> # compute the output