apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-15 05:01:00 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fixed TRT Init bug (#127)

Browse Source 
fixed TRT Init bug
This commit is contained in:
WJJ1995
2022-08-18 13:02:51 +08:00
committed by GitHub
parent de7491f08f
commit ac88f640aa
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/fastdeploy/backends/tensorrt/trt_backend.cc
View File

@@ -259,7 +259,7 @@ bool TrtBackend::InitFromOnnx(const std::string& model_file,
<< option.serialize_file << ", will load it directly." << option.serialize_file << ", will load it directly."
<< std::endl; << std::endl;
fin.close(); fin.close();
return InitFromTrt(option.serialize_file); return InitFromTrt(option.serialize_file, option);
} }
} }