From 899dfde6f545faa73c0ffe5732ae641b10dfbc09 Mon Sep 17 00:00:00 2001 From: Sergio Date: Fri, 3 Oct 2014 17:14:20 -0700 Subject: [PATCH] Adapt lenet_multistep_solver.prototxt to current solvers --- examples/{lenet => mnist}/lenet_multistep_solver.prototxt | 13 +++++-------- examples/mnist/readme.md | 3 +++ 2 files changed, 8 insertions(+), 8 deletions(-) rename examples/{lenet => mnist}/lenet_multistep_solver.prototxt (74%) 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 --- 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 33ed371..ef7f5da 100644 --- a/examples/mnist/readme.md +++ b/examples/mnist/readme.md @@ -282,3 +282,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 -- 2.7.4