diff options
author | Yangqing Jia <jiayq@google.com> | 2014-07-26 08:37:05 -0700 |
---|---|---|
committer | Yangqing Jia <jiayq@google.com> | 2014-07-26 08:37:05 -0700 |
commit | 9862d5a4dcc70edba94291a64c303b659e7bd362 (patch) | |
tree | c006bc0c3b11842c1176b1e864944042139a1bad /examples/mnist | |
parent | 70b01f2164dd5e53605fc58c25530da3a5cccbd3 (diff) | |
download | caffeonacl-9862d5a4dcc70edba94291a64c303b659e7bd362.tar.gz caffeonacl-9862d5a4dcc70edba94291a64c303b659e7bd362.tar.bz2 caffeonacl-9862d5a4dcc70edba94291a64c303b659e7bd362.zip |
LOG(ERROR)->LOG(FATAL), and misc script changes.
Diffstat (limited to 'examples/mnist')
-rw-r--r-- | examples/mnist/readme.md | 2 | ||||
-rwxr-xr-x | examples/mnist/train_lenet.sh | 3 | ||||
-rwxr-xr-x | examples/mnist/train_lenet_consolidated.sh | 3 | ||||
-rwxr-xr-x | examples/mnist/train_mnist_autoencoder.sh | 3 |
4 files changed, 7 insertions, 4 deletions
diff --git a/examples/mnist/readme.md b/examples/mnist/readme.md index d609cfff..4f3f4d9c 100644 --- a/examples/mnist/readme.md +++ b/examples/mnist/readme.md @@ -212,7 +212,7 @@ Training the model is simple after you have written the network definition proto cd $CAFFE_ROOT/examples/mnist ./train_lenet.sh -`train_lenet.sh` is a simple script, but here are a few explanations: `GLOG_logtostderr=1` is the google logging flag that prints all the logging messages directly to stderr. The main tool for training is `train_net.bin`, with the solver protobuf text file as its argument. +`train_lenet.sh` is a simple script, but here are a few explanations: `GLOG_logtostderr=1` is the google logging flag that prints all the logging messages directly to stderr. The main tool for training is `caffe.bin` with action `train`, with the solver protobuf text file as its argument. When you run the code, you will see a lot of messages flying by like this: diff --git a/examples/mnist/train_lenet.sh b/examples/mnist/train_lenet.sh index c30fc3e0..4d0f1eea 100755 --- a/examples/mnist/train_lenet.sh +++ b/examples/mnist/train_lenet.sh @@ -2,4 +2,5 @@ TOOLS=../../build/tools -GLOG_logtostderr=1 $TOOLS/train_net.bin lenet_solver.prototxt +GLOG_logtostderr=1 $TOOLS/caffe.bin train \ + --solver_proto_file=lenet_solver.prototxt diff --git a/examples/mnist/train_lenet_consolidated.sh b/examples/mnist/train_lenet_consolidated.sh index ff6897e2..802c801e 100755 --- a/examples/mnist/train_lenet_consolidated.sh +++ b/examples/mnist/train_lenet_consolidated.sh @@ -2,4 +2,5 @@ TOOLS=../../build/tools -GLOG_logtostderr=1 $TOOLS/train_net.bin lenet_consolidated_solver.prototxt +GLOG_logtostderr=1 $TOOLS/caffe.bin train \ + --solver_proto_file=lenet_consolidated_solver.prototxt diff --git a/examples/mnist/train_mnist_autoencoder.sh b/examples/mnist/train_mnist_autoencoder.sh index af2245e0..3fb49353 100755 --- a/examples/mnist/train_mnist_autoencoder.sh +++ b/examples/mnist/train_mnist_autoencoder.sh @@ -1,4 +1,5 @@ #!/bin/bash TOOLS=../../build/tools -GLOG_logtostderr=1 $TOOLS/train_net.bin mnist_autoencoder_solver.prototxt +GLOG_logtostderr=1 $TOOLS/caffe.bin train \ + --solver_proto_file=mnist_autoencoder_solver.prototxt |