diff options
author | Ronghang Hu <huronghang@hotmail.com> | 2015-09-17 13:26:30 -0700 |
---|---|---|
committer | Ronghang Hu <huronghang@hotmail.com> | 2015-09-17 13:26:30 -0700 |
commit | 2a585f78a0f5a35e5bb2489969094fd420a986ec (patch) | |
tree | 97962d52cc85e83e004b9fa10a6e36e148f4e2aa /tools | |
parent | 5ff3734fefe5a432c62df81c9370df41638d8d60 (diff) | |
parent | 3d3a8b2ca09b64d94836652d0c9b5ffbb31551f6 (diff) | |
download | caffeonacl-2a585f78a0f5a35e5bb2489969094fd420a986ec.tar.gz caffeonacl-2a585f78a0f5a35e5bb2489969094fd420a986ec.tar.bz2 caffeonacl-2a585f78a0f5a35e5bb2489969094fd420a986ec.zip |
Merge pull request #3074 from ronghanghu/show-use-cpu
Get back 'USE CPU' print for caffe train
Diffstat (limited to 'tools')
-rw-r--r-- | tools/caffe.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tools/caffe.cpp b/tools/caffe.cpp index ff63860a..e3f684b5 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; |