Skip to content

Commit 75fa2a6

Browse files
committed
Update model.
1 parent cc2def8 commit 75fa2a6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

torchvision/models/detection/retinanet.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -560,7 +560,7 @@ def forward(self, images, targets=None):
560560
# TODO: replace with pytorch links
561561
model_urls = {
562562
'retinanet_mobilenet_v3_large_fpn_coco':
563-
'https://github.com/datumbox/torchvision-models/raw/main/retinanet_mobilenet_v3_large_fpn-9385e81f.pth',
563+
'https://github.com/datumbox/torchvision-models/raw/main/retinanet_mobilenet_v3_large_fpn-730b1422.pth',
564564
'retinanet_resnet50_fpn_coco':
565565
'https://download.pytorch.org/models/retinanet_resnet50_fpn_coco-eeacb38b.pth',
566566
}
@@ -648,9 +648,9 @@ def retinanet_mobilenet_v3_large_fpn(pretrained=False, progress=True, num_classe
648648
trainable_backbone_layers (int): number of trainable (not frozen) resnet layers starting from final block.
649649
Valid values are between 0 and 6, with 6 meaning all backbone layers are trainable.
650650
"""
651-
# check default parameters and by default set it to 2 if possible
651+
# check default parameters and by default set it to 3 if possible
652652
trainable_backbone_layers = _validate_trainable_layers(
653-
pretrained or pretrained_backbone, trainable_backbone_layers, 6, 2)
653+
pretrained or pretrained_backbone, trainable_backbone_layers, 6, 3)
654654

655655
if pretrained:
656656
pretrained_backbone = False

0 commit comments

Comments
 (0)