From 5cb7c233671d4dbd15785051287f6317d50b8a19 Mon Sep 17 00:00:00 2001 From: Ross Girshick Date: Sun, 12 Jan 2014 07:54:35 -0800 Subject: 10x learning rate for fine tuning makes a big difference --- models/pascal_finetune.prototxt | 8 ++++---- models/pascal_finetune_solver.prototxt | 2 +- models/pascal_finetune_val.prototxt | 8 ++++---- 3 files changed, 9 insertions(+), 9 deletions(-) (limited to 'models') 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. } -- cgit v1.2.3