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; }