From 02899f9a1afddbb0841114e418657942821f45da Mon Sep 17 00:00:00 2001 From: Moe Date: Thu, 20 Jan 2022 22:20:10 +0330 Subject: [PATCH] fix docstring by removing ResNet --- torchvision/models/detection/backbone_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/detection/backbone_utils.py b/torchvision/models/detection/backbone_utils.py index 54fdc4c05c9..808c69dfe73 100644 --- a/torchvision/models/detection/backbone_utils.py +++ b/torchvision/models/detection/backbone_utils.py @@ -84,7 +84,7 @@ def resnet_fpn_backbone( >>> ('pool', torch.Size([1, 256, 1, 1]))] Args: - backbone_name (string): resnet architecture. Possible values are 'ResNet', 'resnet18', 'resnet34', 'resnet50', + backbone_name (string): resnet architecture. Possible values are 'resnet18', 'resnet34', 'resnet50', 'resnet101', 'resnet152', 'resnext50_32x4d', 'resnext101_32x8d', 'wide_resnet50_2', 'wide_resnet101_2' pretrained (bool): If True, returns a model with backbone pre-trained on Imagenet norm_layer (callable): it is recommended to use the default value. For details visit: