From aff92f87ea4fa234676a406c63f2e6accae5e67a Mon Sep 17 00:00:00 2001 From: Hu Ye Date: Tue, 18 May 2021 17:17:57 +0800 Subject: [PATCH] Create retinanet_swint_T_FPN_3x_.yaml --- configs/SwinT/retinanet_swint_T_FPN_3x_.yaml | 21 ++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 configs/SwinT/retinanet_swint_T_FPN_3x_.yaml diff --git a/configs/SwinT/retinanet_swint_T_FPN_3x_.yaml b/configs/SwinT/retinanet_swint_T_FPN_3x_.yaml new file mode 100644 index 0000000..19a09c0 --- /dev/null +++ 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",)