Fixups per review comments.

BVLC · Aug 16, 2015 · bf2c331 · bf2c331
1 parent f4dc253
commit bf2c331
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/include/caffe/solver.hpp b/include/caffe/solver.hpp
@@ -77,8 +77,6 @@ class Solver {
   string SnapshotToBinaryProto();
   string SnapshotToHDF5();
   // The test routine
-  // stop_was_requested will be set to true iff a request to stop training
-  // was received whilst testing.
   void TestAll();
   void Test(const int test_net_id = 0);
   virtual void SnapshotSolverState(const string& model_filename) = 0;

diff --git a/src/caffe/util/signal_handler.cpp b/src/caffe/util/signal_handler.cpp
@@ -1,6 +1,4 @@
 #include <boost/bind.hpp>
-#include <boost/thread/mutex.hpp>
-#include <boost/thread/thread.hpp>
 #include <glog/logging.h>
 
 #include <signal.h>

diff --git a/tools/caffe.cpp b/tools/caffe.cpp
@@ -133,8 +133,10 @@ void CopyLayers(caffe::Solver<float>* solver, const std::string& model_list) {
   }
 }
 
+// Translate the signal effect the user specified on the command-line to the
+// corresponding enumeration.
 caffe::SolverParameter_Action GetRequestedAction(
-                              const std::string& flag_value) {
+    const std::string& flag_value) {
   if (flag_value == "stop") {
     return caffe::SolverParameter_Action_STOP;
   }