From 8eb05e1f47860cc18b8af2ed5ebc8d5169c03eaf Mon Sep 17 00:00:00 2001 From: Evan Shelhamer Date: Wed, 6 Aug 2014 19:45:05 -0700 Subject: [PATCH] consolidaet GPU flag for caffe cli --- tools/caffe.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/caffe.cpp b/tools/caffe.cpp index 1672b60..26db178 100644 --- a/tools/caffe.cpp +++ b/tools/caffe.cpp @@ -32,8 +32,8 @@ DEFINE_string(pretrained_net_file, "", "Cannot be set simultaneously with resume_point_file."); DEFINE_int32(run_iterations, 50, "[time] The number of iterations to run."); -DEFINE_bool(time_with_gpu, false, - "[time] Test the model with GPU."); +DEFINE_bool(gpu, false, + "[time] Run in GPU mode when true."); // A simple registry for caffe commands. typedef int (*BrewFunction)(); @@ -103,7 +103,7 @@ RegisterBrewFunction(train); int time() { // Set device id and mode - if (FLAGS_time_with_gpu) { + if (FLAGS_gpu) { LOG(INFO) << "Use GPU with device id " << FLAGS_device_id; Caffe::SetDevice(FLAGS_device_id); Caffe::set_mode(Caffe::GPU); -- 2.7.4