From f9566976ecb06cd44aa8a76cb5cb308d73a57a4e Mon Sep 17 00:00:00 2001 From: wang-xinyu Date: Thu, 5 Nov 2020 20:41:07 +0800 Subject: [PATCH] fix --- mobilenetv3/mobilenet_v3.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/mobilenetv3/mobilenet_v3.cpp b/mobilenetv3/mobilenet_v3.cpp index ff7198c..19eeeb9 100644 --- a/mobilenetv3/mobilenet_v3.cpp +++ b/mobilenetv3/mobilenet_v3.cpp @@ -444,9 +444,8 @@ int main(int argc, char** argv) APIToModel(1, &modelStream, mode); assert(modelStream != nullptr); - std::ofstream p("mobilenetv3_" + mode + ".engine"); - if (!p) - { + std::ofstream p("mobilenetv3_" + mode + ".engine", std::ios::binary); + if (!p) { std::cerr << "could not open plan output file" << std::endl; return -1; }