Make tools/ for core binaries, stow scripts/ in tools/extra

Collect core Caffe tools like train_net, device_query, etc. together in
tools/ and include helper scripts under tools/extra.

1 files changed, 37 insertions, 0 deletions

diff --git a/tools/train_net.cpp b/tools/train_net.cpp
new file mode 100644
index 00000000..ce62616b
--- /dev/null
+++ b/tools/train_net.cpp
@@ -0,0 +1,37 @@
+// Copyright 2013 Yangqing Jia
+//
+// This is a simple script that allows one to quickly train a network whose
+// parameters are specified by text format protocol buffers.
+// Usage:
+//    train_net net_proto_file solver_proto_file [resume_point_file]
+
+#include <cuda_runtime.h>
+
+#include <cstring>
+
+#include "caffe/caffe.hpp"
+
+using namespace caffe;
+
+int main(int argc, char** argv) {
+  ::google::InitGoogleLogging(argv[0]);
+  if (argc < 2) {
+    LOG(ERROR) << "Usage: train_net solver_proto_file [resume_point_file]";
+    return 0;
+  }
+
+  SolverParameter solver_param;
+  ReadProtoFromTextFile(argv[1], &solver_param);
+
+  LOG(INFO) << "Starting Optimization";
+  SGDSolver<float> solver(solver_param);
+  if (argc == 3) {
+    LOG(INFO) << "Resuming from " << argv[2];
+    solver.Solve(argv[2]);
+  } else {
+    solver.Solve();
+  }
+  LOG(INFO) << "Optimization Done.";
+
+  return 0;
+}