Skip to content

Commit 8e2084c

Browse files
committed
yolov2: Add yolov2 train solver prototxt and prune cfg
Signed-off-by: Huaqi Fang <[email protected]>
1 parent df2d235 commit 8e2084c

File tree

4 files changed

+1217
-0
lines changed

4 files changed

+1217
-0
lines changed
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
[device]
2+
gpuid = 0
3+
4+
[net_params]
5+
supported_batchsize = 15
6+
7+
[filenames]
8+
solver_filename = yolo_voc_solverprune.prototxt
9+
train_filename = yolo_voc_trainval.prototxt
10+
deploy_filename = yolo_voc_snpscaffe.prototxt
11+
caffemodel_filename = yolo_voc.caffemodel
12+
13+
[Log]
14+
LOG_LEVEL = 1
Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
net: "yolo_voc_trainval.prototxt"
2+
test_iter: 1238
3+
test_interval: 5000000
4+
test_initialization: false
5+
display: 100
6+
average_loss: 100
7+
lr_policy: "step"
8+
base_lr: 0.00001
9+
gamma: 0.8
10+
stepsize: 10000
11+
max_iter: 100000
12+
momentum: 0.9
13+
weight_decay: 0.0005
14+
snapshot: 5000
15+
snapshot_prefix: "snapshot/"
16+
#type: "Adam"
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
net: "yolo_voc_trainval.prototxt"
2+
test_iter: 1238
3+
test_interval: 400000
4+
test_initialization: false
5+
display: 100
6+
average_loss: 100
7+
lr_policy: "step"
8+
base_lr: 0.00001
9+
gamma: 0.8
10+
stepsize: 30000
11+
max_iter: 60000
12+
momentum: 0.9
13+
weight_decay: 0.0005
14+
snapshot: 30000
15+
snapshot_prefix: "snapshot/"

0 commit comments

Comments
 (0)