summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRonghang Hu <huronghang@hotmail.com>2015-09-17 20:26:30 (GMT)
committerRonghang Hu <huronghang@hotmail.com>2015-09-17 20:26:30 (GMT)
commit2a585f78a0f5a35e5bb2489969094fd420a986ec (patch)
tree97962d52cc85e83e004b9fa10a6e36e148f4e2aa
parent5ff3734fefe5a432c62df81c9370df41638d8d60 (diff)
parent3d3a8b2ca09b64d94836652d0c9b5ffbb31551f6 (diff)
downloadcaffeonacl-2a585f78a0f5a35e5bb2489969094fd420a986ec.zip
caffeonacl-2a585f78a0f5a35e5bb2489969094fd420a986ec.tar.gz
caffeonacl-2a585f78a0f5a35e5bb2489969094fd420a986ec.tar.bz2
Merge pull request #3074 from ronghanghu/show-use-cpu
Get back 'USE CPU' print for caffe train
-rw-r--r--tools/caffe.cpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/tools/caffe.cpp b/tools/caffe.cpp
index ff63860..e3f684b 100644
--- 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;