Fix to detr_resnet101_dc5 in torch.hub (#90)

oke-aditya · Dec 12, 2020 · ea2c74b · ea2c74b
1 parent 97ba885
commit ea2c74b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/quickvision/models/detection/detr/model_factory.py b/quickvision/models/detection/detr/model_factory.py
@@ -60,7 +60,7 @@ def create_detr_backbone(model_name: str, pretrained: str = None,):
         backbone = torch.hub.load('facebookresearch/detr', 'detr_resnet50_dc5', pretrained=False)
 
     elif(model_name == "resnet101_dc5"):
-        backbone = torch.hub.load('facebookresearch/detr', 'detr_resnet50_dc5', pretrained=False)
+        backbone = torch.hub.load('facebookresearch/detr', 'detr_resnet101_dc5', pretrained=False)
 
     else:
         raise ValueError("Unuspported backbone")