Merge pull request BVLC#3074 from ronghanghu/show-use-cpu

Get back 'USE CPU' print for caffe train
jihochu · Sep 17, 2015 · 2a585f7 · 2a585f7
2 parents 5ff3734 + 3d3a8b2
commit 2a585f7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/caffe.cpp b/tools/caffe.cpp
@@ -174,6 +174,7 @@ int train() {
   vector<int> gpus;
   get_gpus(&gpus);
   if (gpus.size() == 0) {
+    LOG(INFO) << "Use CPU.";
     Caffe::set_mode(Caffe::CPU);
   } else {
     ostringstream s;