Skip to content

Commit b8e35c6

Browse files
pesserpesser
committed
coco config
1 parent 180a123 commit b8e35c6

File tree

2 files changed

+76
-0
lines changed

2 files changed

+76
-0
lines changed

‎coco.yaml

+38
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
data_index: data/coco/index.p
2+
batch_size: 8
3+
init_batches: 4
4+
spatial_size: 256
5+
box_factor: 2
6+
bottleneck_factor: 2
7+
8+
lr: 1.0e-3
9+
lr_decay_begin: 1000
10+
lr_decay_end: 100000
11+
log_freq: 250
12+
ckpt_freq: 1000
13+
test_freq: 1000
14+
drop_prob: 0.0
15+
16+
feature_layers: [
17+
"input_1",
18+
"block1_conv2",
19+
"block2_conv2",
20+
"block3_conv2",
21+
"block4_conv2",
22+
"block5_conv2"]
23+
24+
feature_weights: [
25+
1.0,
26+
1.0,
27+
1.0,
28+
1.0,
29+
1.0,
30+
1.0]
31+
32+
gram_weights: [
33+
0.0,
34+
0.01,
35+
0.01,
36+
0.01,
37+
0.01,
38+
0.01]

‎coco_retrain.yaml

+38
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
data_index: data/coco/index.p
2+
batch_size: 8
3+
init_batches: 4
4+
spatial_size: 256
5+
box_factor: 2
6+
bottleneck_factor: 2
7+
8+
lr: 1.0e-3
9+
lr_decay_begin: 1000
10+
lr_decay_end: 100000
11+
log_freq: 250
12+
ckpt_freq: 1000
13+
test_freq: 1000
14+
drop_prob: 0.0
15+
16+
feature_layers: [
17+
"input_1",
18+
"block1_conv2",
19+
"block2_conv2",
20+
"block3_conv2",
21+
"block4_conv2",
22+
"block5_conv2"]
23+
24+
feature_weights: [
25+
2.0,
26+
2.0,
27+
2.0,
28+
2.0,
29+
2.0,
30+
2.0]
31+
32+
gram_weights: [
33+
0.0,
34+
0.02,
35+
0.02,
36+
0.02,
37+
0.02,
38+
0.02]

0 commit comments

Comments
 (0)