Adapt lenet_multistep_solver.prototxt to current solvers

diff --git a/examples/lenet/lenet_multistep_solver.prototxt b/examples/mnist/lenet_multistep_solver.prototxt
similarity index 74%
rename from examples/lenet/lenet_multistep_solver.prototxt
rename to examples/mnist/lenet_multistep_solver.prototxt
index fadd7c9..7ee9fb6 100644 (file)
--- a/examples/lenet/lenet_multistep_solver.prototxt
+++ b/examples/mnist/lenet_multistep_solver.prototxt
@@ -1,7 +1,5 @@
-# The training protocol buffer definition
-train_net: "lenet_train.prototxt"
-# The testing protocol buffer definition
-test_net: "lenet_test.prototxt"
+# The train/test net protocol buffer definition
+net: "examples/mnist/lenet_train_test.prototxt"
 # test_iter specifies how many forward passes the test should carry out.
 # In the case of MNIST, we have test batch size 100 and 100 test iterations,
 # covering the full 10,000 testing images.
@@ -27,7 +25,6 @@ display: 100
 max_iter: 10000
 # snapshot intermediate results
 snapshot: 5000
-snapshot_prefix: "lenet"
-# solver mode: 0 for CPU and 1 for GPU
-solver_mode: 1
-device_id: 1
+snapshot_prefix: "examples/mnist/lenet_multistep"
+# solver mode: CPU or GPU
+solver_mode: GPU

diff --git a/examples/mnist/readme.md b/examples/mnist/readme.md
index 44e0091..1f8a869 100644 (file)
--- a/examples/mnist/readme.md
+++ b/examples/mnist/readme.md
@@ -283,3 +283,6 @@ You just did! All the training was carried out on the GPU. In fact, if you would
 and you will be using CPU for training. Isn't that easy?
 
 MNIST is a small dataset, so training with GPU does not really introduce too much benefit due to communication overheads. On larger datasets with more complex models, such as ImageNet, the computation speed difference will be more significant.
+
+### How to reduce the learning rate a fixed steps?
+Look at lenet_multistep_solver.prototxt