haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
wang-xinyu 2020-11-05 20:41:07 +08:00
parent 13ec643933
commit f9566976ec
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mobilenetv3/mobilenet_v3.cpp
View File

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