Skip to content

Commit 06e3e72

Browse files
committed
Adding temp candidate pretrained model for mobilenetv3 large.
1 parent a56fe27 commit 06e3e72

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

torchvision/models/detection/retinanet.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -556,9 +556,10 @@ def forward(self, images, targets=None):
556556
return losses, detections
557557
return self.eager_outputs(losses, detections)
558558

559-
559+
# TODO: replace with pytorch links
560560
model_urls = {
561-
'retinanet_mobilenet_v3_large_fpn_coco': None, # TODO: add pretrained model
561+
'retinanet_mobilenet_v3_large_fpn_coco':
562+
'https://github.com/datumbox/torchvision-models/raw/main/retinanet_mobilenet_v3_large_fpn-9385e81f.pth',
562563
'retinanet_resnet50_fpn_coco':
563564
'https://download.pytorch.org/models/retinanet_resnet50_fpn_coco-eeacb38b.pth',
564565
}

0 commit comments

Comments
 (0)