LOG(ERROR)->LOG(FATAL), and misc script changes.
authorYangqing Jia <jiayq@google.com>
Sat, 26 Jul 2014 15:37:05 +0000 (08:37 -0700)
committerYangqing Jia <jiayq@google.com>
Sat, 26 Jul 2014 15:37:05 +0000 (08:37 -0700)
14 files changed:
examples/cifar10/readme.md
examples/cifar10/train_full.sh
examples/cifar10/train_quick.sh
examples/imagenet/resume_training.sh
examples/imagenet/train_alexnet.sh
examples/imagenet/train_imagenet.sh
examples/mnist/readme.md
examples/mnist/train_lenet.sh
examples/mnist/train_lenet_consolidated.sh
examples/mnist/train_mnist_autoencoder.sh
tools/device_query.cpp
tools/finetune_net.cpp
tools/net_speed_benchmark.cpp
tools/train_net.cpp

index 9d5bd7b..315713f 100644 (file)
@@ -42,7 +42,7 @@ Training the model is simple after you have written the network definition proto
     cd $CAFFE_ROOT/examples/cifar10
     ./train_quick.sh
 
-`train_quick.sh` is a simple script, so have a look inside. `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_quick.sh` is a simple script, so have a look inside. `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 the `train` action, and the solver protobuf text file as its argument.
 
 When you run the code, you will see a lot of messages flying by like this:
 
index 4db7b9a..930b27a 100755 (executable)
@@ -2,15 +2,15 @@
 
 TOOLS=../../build/tools
 
-GLOG_logtostderr=1 $TOOLS/train_net.bin \
-    cifar10_full_solver.prototxt
+GLOG_logtostderr=1 $TOOLS/caffe.bin train \
+    --solver_proto_file=cifar10_full_solver.prototxt
 
 #reduce learning rate by factor of 10
-GLOG_logtostderr=1 $TOOLS/train_net.bin \
-    cifar10_full_solver_lr1.prototxt \
-    cifar10_full_iter_60000.solverstate
+GLOG_logtostderr=1 $TOOLS/caffe.bin train \
+    --solver_proto_file=cifar10_full_solver_lr1.prototxt \
+    --resume_point_file=cifar10_full_iter_60000.solverstate
 
 #reduce learning rate by factor of 10
-GLOG_logtostderr=1 $TOOLS/train_net.bin \
-    cifar10_full_solver_lr2.prototxt \
-    cifar10_full_iter_65000.solverstate
+GLOG_logtostderr=1 $TOOLS/caffe.bin train \
+    --solver_proto_file=cifar10_full_solver_lr2.prototxt \
+    --resume_point_file=cifar10_full_iter_65000.solverstate
index 1d954b5..5e8fffc 100755 (executable)
@@ -2,7 +2,10 @@
 
 TOOLS=../../build/tools
 
-GLOG_logtostderr=1 $TOOLS/train_net.bin cifar10_quick_solver.prototxt
+GLOG_logtostderr=1 $TOOLS/caffe.bin train \
+  --solver_proto_file=cifar10_quick_solver.prototxt
 
 #reduce learning rate by fctor of 10 after 8 epochs
-GLOG_logtostderr=1 $TOOLS/train_net.bin cifar10_quick_solver_lr1.prototxt cifar10_quick_iter_4000.solverstate
+GLOG_logtostderr=1 $TOOLS/caffe.bin train \
+  --solver_proto_file=cifar10_quick_solver_lr1.prototxt \
+  --resume_point_file=cifar10_quick_iter_4000.solverstate
index 2b3b403..b43a9b1 100755 (executable)
@@ -2,7 +2,8 @@
 
 TOOLS=../../build/tools
 
-GLOG_logtostderr=1 $TOOLS/train_net.bin \
-    imagenet_solver.prototxt caffe_imagenet_train_10000.solverstate
+GLOG_logtostderr=1 $TOOLS/caffe.bin train\
+    --solver_proto_file=imagenet_solver.prototxt \
+    --resume_point_file=caffe_imagenet_train_10000.solverstate
 
 echo "Done."
index c91e910..d43a235 100755 (executable)
@@ -2,6 +2,7 @@
 
 TOOLS=../../build/tools
 
-GLOG_logtostderr=1 $TOOLS/train_net.bin alexnet_solver.prototxt
+GLOG_logtostderr=1 $TOOLS/caffe.bin train \
+  --solver_proto_file=alexnet_solver.prototxt
 
 echo "Done."
index 0d2563b..90dcdfa 100755 (executable)
@@ -2,6 +2,7 @@
 
 TOOLS=../../build/tools
 
-GLOG_logtostderr=1 $TOOLS/train_net.bin imagenet_solver.prototxt
+GLOG_logtostderr=1 $TOOLS/caffe.bin train \
+  --solver_proto_file=imagenet_solver.prototxt
 
 echo "Done."
index d609cff..4f3f4d9 100644 (file)
@@ -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:
 
index c30fc3e..4d0f1ee 100755 (executable)
@@ -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
index ff6897e..802c801 100755 (executable)
@@ -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
index af2245e..3fb4935 100755 (executable)
@@ -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
index a87e23c..efdbbbb 100644 (file)
@@ -3,7 +3,7 @@
 #include "caffe/common.hpp"
 
 int main(int argc, char** argv) {
-  LOG(ERROR) << "Deprecated. Use caffe.bin devicequery "
+  LOG(FATAL) << "Deprecated. Use caffe.bin devicequery "
                 "[--device_id=0] instead.";
   return 0;
 }
index 17b7f3a..213ff01 100644 (file)
@@ -3,7 +3,7 @@
 #include "caffe/caffe.hpp"
 
 int main(int argc, char** argv) {
-  LOG(ERROR) << "Deprecated. Use caffe.bin train --solver_proto_file=... "
+  LOG(FATAL) << "Deprecated. Use caffe.bin train --solver_proto_file=... "
                 "[--pretrained_net_file=...] instead.";
   return 0;
 }
index 9e6bb4c..f520e1d 100644 (file)
@@ -3,7 +3,7 @@
 #include "caffe/caffe.hpp"
 
 int main(int argc, char** argv) {
-  LOG(ERROR) << "Deprecated. Use caffe.bin speedtest --net_proto_file=... "
+  LOG(FATAL) << "Deprecated. Use caffe.bin speedtest --net_proto_file=... "
              "[--run_iterations=50] [--speedtest_with_gpu] [--device_id=0]";
   return 0;
 }
index 35b1c77..08602d4 100644 (file)
@@ -3,7 +3,7 @@
 #include "caffe/caffe.hpp"
 
 int main(int argc, char** argv) {
-  LOG(ERROR) << "Deprecated. Use caffe.bin train --solver_proto_file=... "
+  LOG(FATAL) << "Deprecated. Use caffe.bin train --solver_proto_file=... "
                 "[--resume_point_file=...] instead.";
   return 0;
 }