caffe

1
net: "models/finetune_flickr_style/train_val.prototxt"
2
test_iter: 100
3
test_interval: 1000
4
# lr for fine-tuning should be lower than when starting from scratch
5
base_lr: 0.001
6
lr_policy: "step"
7
gamma: 0.1
8
# stepsize should also be lower, as we're closer to being done
9
stepsize: 20000
10
display: 20
11
max_iter: 100000
12
momentum: 0.9
13
weight_decay: 0.0005
14
snapshot: 10000
15
snapshot_prefix: "models/finetune_flickr_style/finetune_flickr_style"
16
# uncomment the following to default to CPU mode solving
17
# solver_mode: CPU
18