diff options
| author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-08-06 19:45:05 -0700 |
|---|---|---|
| committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-08-06 23:44:26 -0700 |
| commit | 8eb05e1f47860cc18b8af2ed5ebc8d5169c03eaf (patch) | |
| tree | 6bb3194e238a230b3929b9bdce6d69646563c455 /tools/caffe.cpp | |
| parent | de6123c4f4f0e24220a874247971d452ae43874e (diff) | |
| download | caffeonacl-8eb05e1f47860cc18b8af2ed5ebc8d5169c03eaf.tar.gz caffeonacl-8eb05e1f47860cc18b8af2ed5ebc8d5169c03eaf.tar.bz2 caffeonacl-8eb05e1f47860cc18b8af2ed5ebc8d5169c03eaf.zip | |
consolidaet GPU flag for caffe cli
Diffstat (limited to 'tools/caffe.cpp')
| -rw-r--r-- | tools/caffe.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tools/caffe.cpp b/tools/caffe.cpp index 1672b600..26db1785 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); |