add the config of stage 1

sunke123 · web-flow · commit 530c6013d729 · 2019-08-28T16:24:38.000+08:00
diff --git a/experiments/cls_hrnet_w18_sgd_lr5e-2_wd1e-4_bs32_x100.yaml b/experiments/cls_hrnet_w18_sgd_lr5e-2_wd1e-4_bs32_x100.yaml
@@ -11,6 +11,15 @@ MODEL:
     - 224
     - 224
   EXTRA:
+    STAGE1:
+      NUM_MODULES: 1
+      NUM_RANCHES: 1
+      BLOCK: BOTTLENECK
+      NUM_BLOCKS:
+      - 4
+      NUM_CHANNELS:
+      - 64
+      FUSE_METHOD: SUM
     STAGE2:
       NUM_MODULES: 1
       NUM_BRANCHES: 2
