summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorRonghang Hu <huronghang@hotmail.com>2015-09-16 19:06:16 (GMT)
committerRonghang Hu <huronghang@hotmail.com>2015-09-16 19:06:16 (GMT)
commit3d3a8b2ca09b64d94836652d0c9b5ffbb31551f6 (patch)
tree178e083ea7ffa05f58487ba3da8fb4bcf3cb4e9d
parent71e05876f644a08af4cb1c955d01c5a272539e96 (diff)
downloadcaffeonacl-3d3a8b2ca09b64d94836652d0c9b5ffbb31551f6.zip
caffeonacl-3d3a8b2ca09b64d94836652d0c9b5ffbb31551f6.tar.gz
caffeonacl-3d3a8b2ca09b64d94836652d0c9b5ffbb31551f6.tar.bz2
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;