From 3d3a8b2ca09b64d94836652d0c9b5ffbb31551f6 Mon Sep 17 00:00:00 2001 From: Ronghang Hu Date: Wed, 16 Sep 2015 12:06:16 -0700 Subject: [PATCH] Get back 'USE CPU' print for caffe train --- tools/caffe.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/caffe.cpp b/tools/caffe.cpp index ff63860a3c1..e3f684b5ab3 100644 --- a/tools/caffe.cpp +++ b/tools/caffe.cpp @@ -174,6 +174,7 @@ int train() { vector gpus; get_gpus(&gpus); if (gpus.size() == 0) { + LOG(INFO) << "Use CPU."; Caffe::set_mode(Caffe::CPU); } else { ostringstream s;