Create retinanet_swint_T_FPN_3x_.yaml

xiaohu2015 · May 18, 2021 · aff92f8 · aff92f8
1 parent 466321c
commit aff92f8
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/configs/SwinT/retinanet_swint_T_FPN_3x_.yaml b/configs/SwinT/retinanet_swint_T_FPN_3x_.yaml
@@ -0,0 +1,21 @@
+_BASE_: "../Base-RetinaNet.yaml"
+MODEL:
+  WEIGHTS: "swin_tiny_patch4_window7_224_d2.pth"
+  RESNETS:
+    DEPTH: 50
+  BACKBONE:
+    NAME: "build_retinanet_swint_fpn_backbone"
+  SWINT:
+    OUT_FEATURES: ["stage3", "stage4", "stage5"]
+  FPN:
+    IN_FEATURES: ["stage3", "stage4", "stage5"]
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000
+  WEIGHT_DECAY: 0.05
+  BASE_LR: 0.0001
+  AMP:
+    ENABLED: True
+DATASETS:
+  TRAIN: ("coco_2017_train",)
+  TEST: ("coco_2017_val",)