fix

wang-xinyu · wang-xinyu · commit f9566976ecb0 · 2020-11-06T23:55:43.000+08:00
diff --git 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;
         }
