summaryrefslogtreecommitdiff
path: root/models
diff options
context:
space:
mode:
authorRoss Girshick <rbg@eecs.berkeley.edu>2014-01-12 07:54:35 -0800
committerEvan Shelhamer <shelhamer@imaginarynumber.net>2014-03-19 19:07:23 -0700
commit5cb7c233671d4dbd15785051287f6317d50b8a19 (patch)
treee2111ee6f009aafcdc35994b4f220e2bedf1e29d /models
parent17a0c1686b1de328765b94e44d2636c6cb15caf4 (diff)
downloadcaffeonacl-5cb7c233671d4dbd15785051287f6317d50b8a19.tar.gz
caffeonacl-5cb7c233671d4dbd15785051287f6317d50b8a19.tar.bz2
caffeonacl-5cb7c233671d4dbd15785051287f6317d50b8a19.zip
10x learning rate for fine tuning makes a big difference
Diffstat (limited to 'models')
-rw-r--r--models/pascal_finetune.prototxt8
-rw-r--r--models/pascal_finetune_solver.prototxt2
-rw-r--r--models/pascal_finetune_val.prototxt8
3 files changed, 9 insertions, 9 deletions
diff --git a/models/pascal_finetune.prototxt b/models/pascal_finetune.prototxt
index 229898eb..d2a33f7f 100644
--- a/models/pascal_finetune.prototxt
+++ b/models/pascal_finetune.prototxt
@@ -3,8 +3,8 @@ layers {
layer {
name: "data"
type: "window_data"
- source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_trainval.txt"
- meanfile: "/home/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto"
+ source: "/work4/rbg/convnet-selective-search/selective-search-data/window_file_2007_trainval.txt"
+ meanfile: "/home/eecs/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto"
batchsize: 128
cropsize: 227
context_pad: 16
@@ -350,8 +350,8 @@ layers {
type: "constant"
value: 0
}
- blobs_lr: 100.
- blobs_lr: 200.
+ blobs_lr: 10.
+ blobs_lr: 20.
weight_decay: 1.
weight_decay: 0.
can_clobber: false
diff --git a/models/pascal_finetune_solver.prototxt b/models/pascal_finetune_solver.prototxt
index f53a1d5c..8c464754 100644
--- a/models/pascal_finetune_solver.prototxt
+++ b/models/pascal_finetune_solver.prototxt
@@ -2,7 +2,7 @@ train_net: "examples/pascal_finetune.prototxt"
test_net: "examples/pascal_finetune_val.prototxt"
test_iter: 100
test_interval: 1000
-base_lr: 0.0001
+base_lr: 0.001
lr_policy: "step"
gamma: 0.1
stepsize: 20000
diff --git a/models/pascal_finetune_val.prototxt b/models/pascal_finetune_val.prototxt
index 53aab096..18a680fc 100644
--- a/models/pascal_finetune_val.prototxt
+++ b/models/pascal_finetune_val.prototxt
@@ -3,8 +3,8 @@ layers {
layer {
name: "data"
type: "window_data"
- source: "/work5/rbg/convnet-selective-search/selective-search-data/window_file_2007_test.txt"
- meanfile: "/home/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto"
+ source: "/work4/rbg/convnet-selective-search/selective-search-data/window_file_2007_test.txt"
+ meanfile: "/home/eecs/rbg/working/caffe-rbg/data/ilsvrc2012_mean.binaryproto"
batchsize: 128
cropsize: 227
context_pad: 16
@@ -350,8 +350,8 @@ layers {
type: "constant"
value: 0
}
- blobs_lr: 100.
- blobs_lr: 200.
+ blobs_lr: 10.
+ blobs_lr: 20.
weight_decay: 1.
weight_decay: 0.
}