提交 78c5b9a0 编写于 作者: M MRXLT

bug fix

上级 b4c83b9e
......@@ -248,6 +248,7 @@ class FluidGpuAnalysisDirCore : public FluidFamilyCore {
paddle::AnalysisConfig::Precision::kFloat32,
true,
true);
LOG(INFO) << "create TensorRT predictor";
} else {
if (params.enable_memory_optimization()) {
analysis_config.EnableMemoryOptim();
......
......@@ -248,7 +248,7 @@ class Server(object):
def set_gpuid(self, gpuid=0):
self.gpuid = gpuid
def use_trt(self):
def set_trt(self):
self.use_trt = True
def _prepare_engine(self, model_config_paths, device):
......
......@@ -65,7 +65,7 @@ def start_gpu_card_model(index, gpuid, args): # pylint: disable=doc-string-miss
server.set_ir_optimize(ir_optim)
server.set_max_body_size(max_body_size)
if args.use_trt:
server.use_trt()
server.set_trt()
server.load_model_config(model)
server.prepare_server(workdir=workdir, port=port, device=device)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册