Skip to content

Commit 5631f68

Browse files
sunke123sunke123
authored
add configs of two small models
1 parent 70a4bd3 commit 5631f68

2 files changed

Lines changed: 186 additions & 0 deletions

File tree

‎experiments/cls_hrnet_w18_small_v1_sgd_lr5e-2_wd1e-4_bs32_x100.yaml‎

Lines changed: 93 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,93 @@
1+
GPUS: (0,1,2,3)
2+
LOG_DIR: 'log/'
3+
DATA_DIR: ''
4+
OUTPUT_DIR: 'output/'
5+
WORKERS: 4
6+
PRINT_FREQ: 1000
7+
8+
MODEL:
9+
NAME: cls_hrnet
10+
IMAGE_SIZE:
11+
- 224
12+
- 224
13+
EXTRA:
14+
WITH_HEAD: true
15+
STAGE1:
16+
NUM_MODULES: 1
17+
NUM_RANCHES: 1
18+
BLOCK: BOTTLENECK
19+
NUM_BLOCKS:
20+
- 1
21+
NUM_CHANNELS:
22+
- 32
23+
FUSE_METHOD: SUM
24+
STAGE2:
25+
NUM_MODULES: 1
26+
NUM_BRANCHES: 2
27+
BLOCK: BASIC
28+
NUM_BLOCKS:
29+
- 2
30+
- 2
31+
NUM_CHANNELS:
32+
- 16
33+
- 32
34+
FUSE_METHOD: SUM
35+
STAGE3:
36+
NUM_MODULES: 1
37+
NUM_BRANCHES: 3
38+
BLOCK: BASIC
39+
NUM_BLOCKS:
40+
- 2
41+
- 2
42+
- 2
43+
NUM_CHANNELS:
44+
- 16
45+
- 32
46+
- 64
47+
FUSE_METHOD: SUM
48+
STAGE4:
49+
NUM_MODULES: 1
50+
NUM_BRANCHES: 4
51+
BLOCK: BASIC
52+
NUM_BLOCKS:
53+
- 2
54+
- 2
55+
- 2
56+
- 2
57+
NUM_CHANNELS:
58+
- 16
59+
- 32
60+
- 64
61+
- 128
62+
FUSE_METHOD: SUM
63+
CUDNN:
64+
BENCHMARK: true
65+
DETERMINISTIC: false
66+
ENABLED: true
67+
DATASET:
68+
DATASET: 'imagenet'
69+
DATA_FORMAT: 'jpg'
70+
ROOT: 'data/imagenet/'
71+
TEST_SET: 'val'
72+
TRAIN_SET: 'train'
73+
TEST:
74+
BATCH_SIZE_PER_GPU: 32
75+
MODEL_FILE: ''
76+
TRAIN:
77+
BATCH_SIZE_PER_GPU: 32
78+
BEGIN_EPOCH: 0
79+
END_EPOCH: 100
80+
RESUME: true
81+
LR_FACTOR: 0.1
82+
LR_STEP:
83+
- 30
84+
- 60
85+
- 90
86+
OPTIMIZER: sgd
87+
LR: 0.05
88+
WD: 0.0001
89+
MOMENTUM: 0.9
90+
NESTEROV: true
91+
SHUFFLE: true
92+
DEBUG:
93+
DEBUG: false

‎experiments/cls_hrnet_w18_small_v2_sgd_lr5e-2_wd1e-4_bs32_x100.yaml‎

Lines changed: 93 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,93 @@
1+
GPUS: (0,1,2,3)
2+
LOG_DIR: 'log/'
3+
DATA_DIR: ''
4+
OUTPUT_DIR: 'output/'
5+
WORKERS: 4
6+
PRINT_FREQ: 1000
7+
8+
MODEL:
9+
NAME: cls_hrnet
10+
IMAGE_SIZE:
11+
- 224
12+
- 224
13+
EXTRA:
14+
WITH_HEAD: true
15+
STAGE1:
16+
NUM_MODULES: 1
17+
NUM_RANCHES: 1
18+
BLOCK: BOTTLENECK
19+
NUM_BLOCKS:
20+
- 2
21+
NUM_CHANNELS:
22+
- 64
23+
FUSE_METHOD: SUM
24+
STAGE2:
25+
NUM_MODULES: 1
26+
NUM_BRANCHES: 2
27+
BLOCK: BASIC
28+
NUM_BLOCKS:
29+
- 2
30+
- 2
31+
NUM_CHANNELS:
32+
- 18
33+
- 36
34+
FUSE_METHOD: SUM
35+
STAGE3:
36+
NUM_MODULES: 3
37+
NUM_BRANCHES: 3
38+
BLOCK: BASIC
39+
NUM_BLOCKS:
40+
- 2
41+
- 2
42+
- 2
43+
NUM_CHANNELS:
44+
- 18
45+
- 36
46+
- 72
47+
FUSE_METHOD: SUM
48+
STAGE4:
49+
NUM_MODULES: 2
50+
NUM_BRANCHES: 4
51+
BLOCK: BASIC
52+
NUM_BLOCKS:
53+
- 2
54+
- 2
55+
- 2
56+
- 2
57+
NUM_CHANNELS:
58+
- 18
59+
- 36
60+
- 72
61+
- 144
62+
FUSE_METHOD: SUM
63+
CUDNN:
64+
BENCHMARK: true
65+
DETERMINISTIC: false
66+
ENABLED: true
67+
DATASET:
68+
DATASET: 'imagenet'
69+
DATA_FORMAT: 'zip'
70+
ROOT: 'data/imagenet/'
71+
TEST_SET: 'val'
72+
TRAIN_SET: 'train'
73+
TEST:
74+
BATCH_SIZE_PER_GPU: 32
75+
MODEL_FILE: ''
76+
TRAIN:
77+
BATCH_SIZE_PER_GPU: 32
78+
BEGIN_EPOCH: 0
79+
END_EPOCH: 100
80+
RESUME: true
81+
LR_FACTOR: 0.1
82+
LR_STEP:
83+
- 30
84+
- 60
85+
- 90
86+
OPTIMIZER: sgd
87+
LR: 0.05
88+
WD: 0.0001
89+
MOMENTUM: 0.9
90+
NESTEROV: true
91+
SHUFFLE: true
92+
DEBUG:
93+
DEBUG: false

0 commit comments

Comments
 (0)