fix paddle backend

This commit is contained in:
wwbitejotunn
2023-02-09 05:51:30 +00:00
committed by Wang Bojun
parent 4742f97819
commit c2e5f6317e

View File

@@ -44,6 +44,7 @@ void PaddleBackend::BuildOption(const PaddleBackendOption& option) {
"file will save to the directory where paddle model saved." "file will save to the directory where paddle model saved."
<< std::endl; << std::endl;
use_static = true; use_static = true;
config_.SetOptimCacheDir(option.trt_option.serialize_file);
} }
config_.EnableTensorRtEngine(option.trt_option.max_workspace_size, config_.EnableTensorRtEngine(option.trt_option.max_workspace_size,
option.trt_option.max_batch_size, 3, option.trt_option.max_batch_size, 3,