Merge branch 'master' of github.com:cheeyos/caffenet into lane

brodyh · Nov 4, 2014 · 1e82b42 · 1e82b42
2 parents 8e25844 + ba6e1a4
commit 1e82b42
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/driving_img_mean.binaryproto b/driving_img_mean.binaryproto
diff --git a/models/brody/solver.prototxt b/models/brody/solver.prototxt
@@ -1,14 +1,14 @@
 net: "models/brody/train_val_brody.prototxt"
 test_iter: 1000
 test_interval: 1000
-base_lr: 0.00000001
+base_lr: 0.0000001
 lr_policy: "step"
 gamma: 0.1
 stepsize: 100000
 display: 20
 max_iter: 1450000
-momentum: 0.2
+momentum: 0.
 weight_decay: 0.00005
-snapshot: 2000
+snapshot: 10000
 snapshot_prefix: "models/brody/caffe_brody_train"
 solver_mode: GPU
diff --git a/models/brody/train_val_brody.prototxt b/models/brody/train_val_brody.prototxt
@@ -8,7 +8,7 @@ layers {
   data_param {
     source: "/deep/group/driving_data/twangcat/lmdb/driving_img_train"
     backend: LMDB
-    batch_size: 25
+    batch_size: 5
   }
   transform_param {
     mean_file: "driving_img_mean.binaryproto"
@@ -24,7 +24,7 @@ layers {
   data_param {
     source: "/deep/group/driving_data/twangcat/lmdb/driving_label_train"
     backend: LMDB
-    batch_size: 25
+    batch_size: 5
   }
   include: { phase: TRAIN }
 }
@@ -37,7 +37,7 @@ layers {
   data_param {
     source: "/deep/group/driving_data/twangcat/lmdb/driving_img_test"
     backend: LMDB
-    batch_size: 25
+    batch_size: 5
   }
   transform_param {
     mean_file: "driving_img_mean.binaryproto"
@@ -53,7 +53,7 @@ layers {
   data_param {
     source: "/deep/group/driving_data/twangcat/lmdb/driving_label_test"
     backend: LMDB
-    batch_size: 25
+    batch_size: 5
   }
   include: { phase: TEST }
 }