Skip to content

Commit e42406f

Browse files
authored
add the configs of stage 1
1 parent 910a9e6 commit e42406f

1 file changed

Lines changed: 10 additions & 1 deletion

File tree

experiments/cls_hrnet_w48_sgd_lr5e-2_wd1e-4_bs32_x100.yaml

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,15 @@ MODEL:
1111
- 224
1212
- 224
1313
EXTRA:
14+
STAGE1:
15+
NUM_MODULES: 1
16+
NUM_RANCHES: 1
17+
BLOCK: BOTTLENECK
18+
NUM_BLOCKS:
19+
- 4
20+
NUM_CHANNELS:
21+
- 64
22+
FUSE_METHOD: SUM
1423
STAGE2:
1524
NUM_MODULES: 1
1625
NUM_BRANCHES: 2
@@ -80,4 +89,4 @@ TRAIN:
8089
NESTEROV: true
8190
SHUFFLE: true
8291
DEBUG:
83-
DEBUG: false
92+
DEBUG: false

0 commit comments

Comments
 (0)