提交 a527ee30 编写于 作者: Z zhangjun

enableTrt bugfix

上级 f9e3201d
......@@ -124,6 +124,9 @@ class PaddleInferenceEngine : public PaddleEngineBase {
}
if (engine_conf.has_use_trt() && engine_conf.use_trt()) {
if (!engine_conf.has_use_gpu() || !engine_conf.use_gpu()) {
config.EnableUseGpu(2000, FLAGS_gpuid);
}
config.EnableTensorRtEngine(1 << 20,
max_batch,
min_subgraph_size,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册