From f40c30bdd5b5773457a711aa2fd05d46ad9b4eaa Mon Sep 17 00:00:00 2001 From: Ross Girshick Date: Fri, 20 Dec 2013 14:32:55 -0800 Subject: [PATCH] adjustments to try to match the setup for fine tuning with cuda-convnet --- models/pascal_finetune.prototxt | 8 ++++---- models/pascal_finetune_solver.prototxt | 4 ++-- models/pascal_finetune_val.prototxt | 10 +++++----- 3 files changed, 11 insertions(+), 11 deletions(-) diff --git a/models/pascal_finetune.prototxt b/models/pascal_finetune.prototxt index 7b2b05d..1ff9dab 100644 --- 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 --- 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 --- 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. } -- 2.7.4