Get back 'USE CPU' print for caffe train

jxav · Sep 16, 2015 · 3d3a8b2 · 3d3a8b2
1 parent 71e0587
commit 3d3a8b2
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;