diff options
| author | Ross Girshick <rbg@eecs.berkeley.edu> | 2014-01-07 15:04:19 -0800 |
|---|---|---|
| committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-03-19 19:06:05 -0700 |
| commit | bb0e72c817c66c61d422b24cdfec9d3a18ca5171 (patch) | |
| tree | 7d10cd66f3e8db2492893e2a34ba0428828aa8df /models | |
| parent | d98ce8fc48d036ec0508756816f87a5e13e1f10a (diff) | |
| download | caffeonacl-bb0e72c817c66c61d422b24cdfec9d3a18ca5171.tar.gz caffeonacl-bb0e72c817c66c61d422b24cdfec9d3a18ca5171.tar.bz2 caffeonacl-bb0e72c817c66c61d422b24cdfec9d3a18ca5171.zip | |
Code that was used to finetune with reasonable success
Diffstat (limited to 'models')
| -rw-r--r-- | models/pascal_finetune.prototxt | 2 | ||||
| -rw-r--r-- | models/pascal_finetune_solver.prototxt | 6 | ||||
| -rw-r--r-- | models/pascal_finetune_val.prototxt | 2 |
3 files changed, 5 insertions, 5 deletions
diff --git a/models/pascal_finetune.prototxt b/models/pascal_finetune.prototxt index 1ff9dab7..5ee9e8a9 100644 --- a/models/pascal_finetune.prototxt +++ b/models/pascal_finetune.prototxt @@ -3,7 +3,7 @@ layers { layer { name: "data" type: "window_data" - source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_train.txt" + source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_trainval.txt" meanfile: "/home/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto" batchsize: 128 cropsize: 227 diff --git a/models/pascal_finetune_solver.prototxt b/models/pascal_finetune_solver.prototxt index 14925ade..7ea9c5d4 100644 --- a/models/pascal_finetune_solver.prototxt +++ b/models/pascal_finetune_solver.prototxt @@ -1,13 +1,13 @@ train_net: "examples/pascal_finetune.prototxt" test_net: "examples/pascal_finetune_val.prototxt" test_iter: 100 -test_interval: 500 +test_interval: 1000 base_lr: 0.0001 lr_policy: "step" gamma: 0.1 -stepsize: 10000 +stepsize: 20000 display: 20 -max_iter: 400000 +max_iter: 100000 momentum: 0.9 weight_decay: 0.0005 snapshot: 10000 diff --git a/models/pascal_finetune_val.prototxt b/models/pascal_finetune_val.prototxt index 1cea4882..c73cf8f4 100644 --- a/models/pascal_finetune_val.prototxt +++ b/models/pascal_finetune_val.prototxt @@ -3,7 +3,7 @@ layers { layer { name: "data" type: "window_data" - source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_val.txt" + source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_test.txt" meanfile: "/home/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto" batchsize: 128 cropsize: 227 |