patch: fix yaml congif

model_weights/buildings_poc_cfg.yml

@@ -2,7 +2,7 @@ CUDNN_BENCHMARK: false
 DATALOADER:
   ASPECT_RATIO_GROUPING: true
   FILTER_EMPTY_ANNOTATIONS: true
-  NUM_WORKERS: 4
+  NUM_WORKERS: 2
   REPEAT_THRESHOLD: 0.0
   SAMPLER_TRAIN: TrainingSampler
 DATASETS:
@@ -10,8 +10,10 @@ DATASETS:
   PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
   PROPOSAL_FILES_TEST: []
   PROPOSAL_FILES_TRAIN: []
-  TEST: []
-  TRAIN: []
+  TEST:
+  - urban-small_test
+  TRAIN:
+  - urban-small_train
 GLOBAL:
   HACK: 1.0
 INPUT:
@@ -27,6 +29,11 @@ INPUT:
   MAX_SIZE_TRAIN: 1333
   MIN_SIZE_TEST: 800
   MIN_SIZE_TRAIN:
+  - 640
+  - 672
+  - 704
+  - 736
+  - 768
   - 800
   MIN_SIZE_TRAIN_SAMPLING: choice
   RANDOM_FLIP: horizontal
@@ -44,22 +51,26 @@ MODEL:
     OFFSET: 0.0
     SIZES:
     - - 32
-      - 64
-      - 128
-      - 256
-      - 512
+    - - 64
+    - - 128
+    - - 256
+    - - 512
   BACKBONE:
     FREEZE_AT: 2
-    NAME: build_resnet_backbone
-  DEVICE: cpu
+    NAME: build_resnet_fpn_backbone
+  DEVICE: cuda
   FPN:
     FUSE_TYPE: sum
-    IN_FEATURES: []
+    IN_FEATURES:
+    - res2
+    - res3
+    - res4
+    - res5
     NORM: ''
     OUT_CHANNELS: 256
   KEYPOINT_ON: false
   LOAD_PROPOSALS: false
-  MASK_ON: false
+  MASK_ON: true
   META_ARCHITECTURE: GeneralizedRCNN
   PANOPTIC_FPN:
     COMBINE:
@@ -87,11 +98,14 @@ MODEL:
     - false
     - false
     - false
-    DEPTH: 50
+    DEPTH: 101
     NORM: FrozenBN
     NUM_GROUPS: 1
     OUT_FEATURES:
+    - res2
+    - res3
     - res4
+    - res5
     RES2_OUT_CHANNELS: 256
     RES5_DILATION: 1
     STEM_OUT_CHANNELS: 64
@@ -155,11 +169,11 @@ MODEL:
     FC_DIM: 1024
     FED_LOSS_FREQ_WEIGHT_POWER: 0.5
     FED_LOSS_NUM_CLASSES: 50
-    NAME: ''
+    NAME: FastRCNNConvFCHead
     NORM: ''
     NUM_CONV: 0
-    NUM_FC: 0
-    POOLER_RESOLUTION: 14
+    NUM_FC: 2
+    POOLER_RESOLUTION: 7
     POOLER_SAMPLING_RATIO: 0
     POOLER_TYPE: ROIAlignV2
     SMOOTH_L1_BETA: 0.0
@@ -169,18 +183,21 @@ MODEL:
   ROI_HEADS:
     BATCH_SIZE_PER_IMAGE: 512
     IN_FEATURES:
-    - res4
+    - p2
+    - p3
+    - p4
+    - p5
     IOU_LABELS:
     - 0
     - 1
     IOU_THRESHOLDS:
     - 0.5
-    NAME: Res5ROIHeads
+    NAME: StandardROIHeads
     NMS_THRESH_TEST: 0.5
-    NUM_CLASSES: 80
+    NUM_CLASSES: 8
     POSITIVE_FRACTION: 0.25
     PROPOSAL_APPEND_GT: true
-    SCORE_THRESH_TEST: 0.7
+    SCORE_THRESH_TEST: 0.25
   ROI_KEYPOINT_HEAD:
     CONV_DIMS:
     - 512
@@ -204,7 +221,7 @@ MODEL:
     CONV_DIM: 256
     NAME: MaskRCNNConvUpsampleHead
     NORM: ''
-    NUM_CONV: 0
+    NUM_CONV: 4
     POOLER_RESOLUTION: 14
     POOLER_SAMPLING_RATIO: 0
     POOLER_TYPE: ROIAlignV2
@@ -218,7 +235,11 @@ MODEL:
     - -1
     HEAD_NAME: StandardRPNHead
     IN_FEATURES:
-    - res4
+    - p2
+    - p3
+    - p4
+    - p5
+    - p6
     IOU_LABELS:
     - 0
     - -1
@@ -230,9 +251,9 @@ MODEL:
     NMS_THRESH: 0.7
     POSITIVE_FRACTION: 0.5
     POST_NMS_TOPK_TEST: 1000
-    POST_NMS_TOPK_TRAIN: 2000
-    PRE_NMS_TOPK_TEST: 6000
-    PRE_NMS_TOPK_TRAIN: 12000
+    POST_NMS_TOPK_TRAIN: 1000
+    PRE_NMS_TOPK_TEST: 1000
+    PRE_NMS_TOPK_TRAIN: 2000
     SMOOTH_L1_BETA: 0.0
   SEM_SEG_HEAD:
     COMMON_STRIDE: 4
@@ -247,13 +268,13 @@ MODEL:
     NAME: SemSegFPNHead
     NORM: GN
     NUM_CLASSES: 54
-  WEIGHTS: /content/output/model_final.pth
+  WEIGHTS: ./output/model_final.pth
 OUTPUT_DIR: ./output
 SEED: -1
 SOLVER:
   AMP:
     ENABLED: false
-  BASE_LR: 0.001
+  BASE_LR: 0.00025
   BASE_LR_END: 0.0
   BIAS_LR_FACTOR: 1.0
   CHECKPOINT_PERIOD: 5000
@@ -263,16 +284,15 @@ SOLVER:
     ENABLED: false
     NORM_TYPE: 2.0
   GAMMA: 0.1
-  IMS_PER_BATCH: 16
+  IMS_PER_BATCH: 2
   LR_SCHEDULER_NAME: WarmupMultiStepLR
-  MAX_ITER: 40000
+  MAX_ITER: 3000
   MOMENTUM: 0.9
   NESTEROV: false
   NUM_DECAYS: 3
   REFERENCE_WORLD_SIZE: 0
   RESCALE_INTERVAL: false
-  STEPS:
-  - 30000
+  STEPS: []
   WARMUP_FACTOR: 0.001
   WARMUP_ITERS: 1000
   WARMUP_METHOD: linear
@@ -302,4 +322,4 @@ TEST:
     ENABLED: false
     NUM_ITER: 200
 VERSION: 2
-VIS_PERIOD: 0
+VIS_PERIOD: 0