apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 17:17:14 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[TRT] fix tensorrt serialize file problem (#274)

Browse Source 
fix tensorrt serialize file problem
This commit is contained in:
DefTruth
2022-09-23 00:40:17 +08:00
committed by GitHub
parent 625845c7d6
commit 5edd9fccd3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -489,7 +489,7 @@ bool TrtBackend::BuildTrtEngine() {
if (option_.serialize_file != "") {
FDINFO << "Serialize TensorRTEngine to local file "
<< option_.serialize_file << "." << std::endl;
std::ofstream engine_file(option_.serialize_file.c_str());
std::ofstream engine_file(option_.serialize_file.c_str(), std::ios::binary | std::ios::out);
if (!engine_file) {
FDERROR << "Failed to open " << option_.serialize_file << " to write."
<< std::endl;