diff options
Diffstat (limited to 'data')
-rwxr-xr-x | data/get_cifar.sh | 12 | ||||
-rw-r--r-- | data/lenet.prototxt | 2 | ||||
-rw-r--r-- | data/lenet_solver.prototxt | 29 | ||||
-rw-r--r-- | data/lenet_test.prototxt | 2 |
4 files changed, 24 insertions, 21 deletions
diff --git a/data/get_cifar.sh b/data/get_cifar.sh deleted file mode 100755 index 6f42bb09..00000000 --- a/data/get_cifar.sh +++ /dev/null @@ -1,12 +0,0 @@ -#!/usr/bin/env sh -# This scripts downloads the mnist data and unzips it. - -echo "Downloading..." - -wget -q http://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz - -echo "Unzipping..." - -tar xzf cifar-10-binary.tar.gz - -echo "Done." diff --git a/data/lenet.prototxt b/data/lenet.prototxt index 085ed43f..c8a99946 100644 --- a/data/lenet.prototxt +++ b/data/lenet.prototxt @@ -3,7 +3,7 @@ layers { layer { name: "mnist" type: "data" - source: "data/mnist-train-leveldb" + source: "mnist-train-leveldb" batchsize: 64 scale: 0.00390625 } diff --git a/data/lenet_solver.prototxt b/data/lenet_solver.prototxt index d58255b9..d0edc0f0 100644 --- a/data/lenet_solver.prototxt +++ b/data/lenet_solver.prototxt @@ -1,12 +1,27 @@ -train_net: "data/lenet.prototxt" -test_net: "data/lenet_test.prototxt" +# The training protocol buffer definition +train_net: "lenet.prototxt" +# The testing protocol buffer definition +test_net: "lenet_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. +test_iter: 100 +# Carry out testing every 500 training iterations. +test_interval: 500 +# The base learning rate, momentum and the weight decay of the network. base_lr: 0.01 +momentum: 0.9 +weight_decay: 0.0005 +# The learning rate policy lr_policy: "inv" gamma: 0.0001 power: 0.75 +# Display every 100 iterations display: 100 -max_iter: 5000 -momentum: 0.9 -weight_decay: 0.0005 -test_iter: 100 -test_interval: 500
\ No newline at end of file +# The maximum number of iterations +max_iter: 10000 +# snapshot intermediate results +snapshot: 5000 +snapshot_prefix: "lenet" +# solver mode: 0 for CPU and 1 for GPU +solver_mode: 1 diff --git a/data/lenet_test.prototxt b/data/lenet_test.prototxt index fdda4a67..676a2a6a 100644 --- a/data/lenet_test.prototxt +++ b/data/lenet_test.prototxt @@ -3,7 +3,7 @@ layers { layer { name: "mnist" type: "data" - source: "data/mnist-test-leveldb" + source: "mnist-test-leveldb" batchsize: 100 scale: 0.00390625 } |