File tree Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Expand file tree Collapse file tree 1 file changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -199,13 +199,18 @@ def create_model(self,
199199 model: a pytorch nn module
200200 """
201201
202- if pretrained is None :
202+ if pretrained == "resnet50" :
203+ if revision is not None :
204+ warnings .warn (
205+ "Ignoring revision and fine-tuning from ResNet50 MS-COCO checkpoint." )
203206 model = RetinaNetHub (backbone_weights = "COCO_V1" ,
204207 num_classes = self .config .num_classes ,
205208 nms_thresh = self .config .nms_thresh ,
206209 score_thresh = self .config .score_thresh ,
207210 label_dict = self .config .label_dict )
208211 elif pretrained == "dinov3" :
212+ warnings .warn (
213+ "Ignoring revision and fine-tuning from DinoV3 Sat-493M checkpoint." )
209214 model = RetinaNetHub (backbone_weights = "dinov3" ,
210215 num_classes = self .config .num_classes ,
211216 nms_thresh = self .config .nms_thresh ,
You can’t perform that action at this time.
0 commit comments