adjustments to try to match the setup for fine tuning with cuda-convnet

diff --git a/models/pascal_finetune.prototxt b/models/pascal_finetune.prototxt
index 7b2b05d..1ff9dab 100644 (file)
--- a/models/pascal_finetune.prototxt
+++ b/models/pascal_finetune.prototxt
@@ -8,8 +8,8 @@ layers {
     batchsize: 128
     cropsize: 227
     mirror: true
-    det_fg_threshold: 0.6
-    det_bg_threshold: 0.3
+    det_fg_threshold: 0.5
+    det_bg_threshold: 0.5
     det_fg_fraction: 0.25
   }
   top: "data"
@@ -349,8 +349,8 @@ layers {
       type: "constant"
       value: 0
     }
-    blobs_lr: 1.
-    blobs_lr: 2.
+    blobs_lr: 100.
+    blobs_lr: 200.
     weight_decay: 1.
     weight_decay: 0.
     can_clobber: false

diff --git a/models/pascal_finetune_solver.prototxt b/models/pascal_finetune_solver.prototxt
index eaed2b3..14925ad 100644 (file)
--- a/models/pascal_finetune_solver.prototxt
+++ b/models/pascal_finetune_solver.prototxt
@@ -1,11 +1,11 @@
 train_net: "examples/pascal_finetune.prototxt"
 test_net: "examples/pascal_finetune_val.prototxt"
 test_iter: 100
-test_interval: 400
+test_interval: 500
 base_lr: 0.0001
 lr_policy: "step"
 gamma: 0.1
-stepsize: 50000
+stepsize: 10000
 display: 20
 max_iter: 400000
 momentum: 0.9

diff --git a/models/pascal_finetune_val.prototxt b/models/pascal_finetune_val.prototxt
index fe1a269..1cea488 100644 (file)
--- a/models/pascal_finetune_val.prototxt
+++ b/models/pascal_finetune_val.prototxt
@@ -8,9 +8,9 @@ layers {
     batchsize: 128
     cropsize: 227
     mirror: true
-    det_fg_threshold: 0.6
-    det_bg_threshold: 0.3
-    det_fg_fraction: 0.5
+    det_fg_threshold: 0.5
+    det_bg_threshold: 0.5
+    det_fg_fraction: 0.25
   }
   top: "data"
   top: "label"
@@ -349,8 +349,8 @@ layers {
       type: "constant"
       value: 0
     }
-    blobs_lr: 1.
-    blobs_lr: 2.
+    blobs_lr: 100.
+    blobs_lr: 200.
     weight_decay: 1.
     weight_decay: 0.
   }