提交 d347ea68 编写于 作者: N nhzlx

fix comments

上级 f3af90d1
...@@ -62,10 +62,10 @@ include_directories("${PADDLE_LIB}/third_party/boost") ...@@ -62,10 +62,10 @@ include_directories("${PADDLE_LIB}/third_party/boost")
include_directories("${PADDLE_LIB}/third_party/eigen3") include_directories("${PADDLE_LIB}/third_party/eigen3")
if (NOT WIN32) if (NOT WIN32)
if (USE_TENSORRT AND WITH_GPU) if (USE_TENSORRT AND WITH_GPU)
include_directories("${TENSORRT_INCLUDE_DIR}") include_directories("${TENSORRT_INCLUDE_DIR}")
link_directories("${TENSORRT_LIB_DIR}") link_directories("${TENSORRT_LIB_DIR}")
endif() endif()
endif(NOT WIN32) endif(NOT WIN32)
if (NOT WIN32) if (NOT WIN32)
......
...@@ -124,9 +124,7 @@ void Main(bool use_gpu, bool use_trt) { ...@@ -124,9 +124,7 @@ void Main(bool use_gpu, bool use_trt) {
config.device = 0; config.device = 0;
config.max_batch_size = 1; config.max_batch_size = 1;
config.fraction_of_gpu_memory = 0.1; // set by yourself config.fraction_of_gpu_memory = 0.1; // set by yourself
predictor = predictor = CreatePaddlePredictor<paddle::contrib::MixedRTConfig>(config);
CreatePaddlePredictor<paddle::contrib::MixedRTConfig,
PaddleEngineKind::kAutoMixedTensorRT>(config);
} }
VLOG(3) << "begin to process data"; VLOG(3) << "begin to process data";
...@@ -166,7 +164,7 @@ int main(int argc, char** argv) { ...@@ -166,7 +164,7 @@ int main(int argc, char** argv) {
} else if (FLAGS_use_gpu) { } else if (FLAGS_use_gpu) {
paddle::demo::Main(true /*use_gpu*/, false); paddle::demo::Main(true /*use_gpu*/, false);
} else { } else {
paddle::demo::Main(false /* use_gpu*/, false /*use_tensorrt*/); paddle::demo::Main(false /*use_gpu*/, false /*use_tensorrt*/);
} }
return 0; return 0;
} }
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册