Skip to content

Commit

Permalink
fix comments
Browse files Browse the repository at this point in the history
  • Loading branch information
EmmonsCurse committed Jun 14, 2023
1 parent 28482e0 commit 5a13c29
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions deploy/cpp/src/test_seg.cc
Original file line number Diff line number Diff line change
Expand Up @@ -16,12 +16,12 @@
DEFINE_string(model_dir, "", "Directory of the inference model. "
"It constains deploy.yaml and infer models");
DEFINE_string(img_path, "", "Path of the test image.");
DECLARE_string(devices); // 'devices' was already defined in paddle/fluid/inference/io.cc
DECLARE_string(devices); // 'devices' was already defined in paddle/fluid/inference/io.cc: "(devices, "", "The devices to be used which is joined by comma.");"
DEFINE_bool(use_trt, false, "Wether enable TensorRT when use GPU. Defualt: false.");
DEFINE_string(trt_precision, "fp32", "The precision of TensorRT, support fp32, fp16 and int8. Default: fp32");
DEFINE_bool(use_trt_dynamic_shape, false, "Wether enable dynamic shape when use GPU and TensorRT. Defualt: false.");
DEFINE_string(dynamic_shape_path, "", "If set dynamic_shape_path, it read the dynamic shape for TRT.");
DECLARE_bool(use_mkldnn); // 'use_mkldnn' was already defined in paddle/phi/core/flags.cc
DECLARE_bool(use_mkldnn); // 'use_mkldnn' was already defined in paddle/phi/core/flags.cc: "(use_mkldnn, false, "Use MKLDNN to run");"
DEFINE_string(save_dir, "", "Directory of the output image.");

typedef struct YamlConfig {
Expand Down

0 comments on commit 5a13c29

Please sign in to comment.