Add repo_or_dir

yoshitomo-matsubara · yoshitomo-matsubara · commit 9c446854df2b · 2024-12-17T21:50:52.000-08:00
diff --git a/configs/official/coco2017/yoshitomo-matsubara/rrpr2020/ghnd-custom_fasterrcnn_resnet50_fpn_from_fasterrcnn_resnet50_fpn.yaml b/configs/official/coco2017/yoshitomo-matsubara/rrpr2020/ghnd-custom_fasterrcnn_resnet50_fpn_from_fasterrcnn_resnet50_fpn.yaml
@@ -33,6 +33,7 @@ models:
       num_classes: 91
     src_ckpt:
   student_model:
+    repo_or_dir: 'yoshitomo-matsubara/sc2-benchmark'
     key: &student_model_key 'custom_fasterrcnn_resnet_fpn'
     _weights: &student_weights !import_get
       key: 'torchvision.models.detection.faster_rcnn.FasterRCNN_ResNet50_FPN_Weights'
diff --git a/configs/official/coco2017/yoshitomo-matsubara/rrpr2020/ghnd-custom_maskrcnn_resnet50_fpn_from_maskrcnn_resnet50_fpn.yaml b/configs/official/coco2017/yoshitomo-matsubara/rrpr2020/ghnd-custom_maskrcnn_resnet50_fpn_from_maskrcnn_resnet50_fpn.yaml
@@ -33,6 +33,7 @@ models:
       num_classes: 91
     src_ckpt:
   student_model:
+    repo_or_dir: 'yoshitomo-matsubara/sc2-benchmark'
     key: &student_model_key 'custom_maskrcnn_resnet_fpn'
     _weights: &student_weights !import_get
       key: 'torchvision.models.detection.mask_rcnn.MaskRCNN_ResNet50_FPN_Weights'
diff --git a/configs/sample/coco2017/ghnd/custom_fasterrcnn_resnet50_fpn_from_fasterrcnn_resnet50_fpn.yaml b/configs/sample/coco2017/ghnd/custom_fasterrcnn_resnet50_fpn_from_fasterrcnn_resnet50_fpn.yaml
@@ -33,6 +33,7 @@ models:
       num_classes: 91
     src_ckpt:
   student_model:
+    repo_or_dir: 'yoshitomo-matsubara/sc2-benchmark'
     key: &student_model_key 'custom_fasterrcnn_resnet_fpn'
     _weights: &student_weights !import_get
       key: 'torchvision.models.detection.faster_rcnn.FasterRCNN_ResNet50_FPN_Weights'
diff --git a/configs/sample/coco2017/ghnd/custom_maskrcnn_resnet50_fpn_from_maskrcnn_resnet50_fpn.yaml b/configs/sample/coco2017/ghnd/custom_maskrcnn_resnet50_fpn_from_maskrcnn_resnet50_fpn.yaml
@@ -33,6 +33,7 @@ models:
       num_classes: 91
     src_ckpt:
   student_model:
+    repo_or_dir: 'yoshitomo-matsubara/sc2-benchmark'
     key: &student_model_key 'custom_maskrcnn_resnet_fpn'
     _weights: &student_weights !import_get
       key: 'torchvision.models.detection.mask_rcnn.MaskRCNN_ResNet50_FPN_Weights'
