From 5edd9fccd318b2d02bf499ab58895102db451e69 Mon Sep 17 00:00:00 2001 From: DefTruth <31974251+DefTruth@users.noreply.github.com> Date: Fri, 23 Sep 2022 00:40:17 +0800 Subject: [PATCH] [TRT] fix tensorrt serialize file problem (#274) fix tensorrt serialize file problem --- fastdeploy/backends/tensorrt/trt_backend.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastdeploy/backends/tensorrt/trt_backend.cc b/fastdeploy/backends/tensorrt/trt_backend.cc index d8c336511..d9efbbedf 100644 --- a/fastdeploy/backends/tensorrt/trt_backend.cc +++ b/fastdeploy/backends/tensorrt/trt_backend.cc @@ -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;