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

Fix issue 1405 for yolov7 (#1484)

Browse Source
This commit is contained in:
Zhang-Wen-chao 2024-04-10 18:26:05 +08:00 committed by GitHub
parent 62050de3e7
commit c889b84df2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
yolov7/main.cpp
View File

@ -44,9 +44,9 @@ void serialize_engine(unsigned int maxBatchSize, std::string& wts_name, std::str
} }
p.write(reinterpret_cast<const char*>(serialized_engine->data()), serialized_engine->size()); p.write(reinterpret_cast<const char*>(serialized_engine->data()), serialized_engine->size());
delete builder;
delete config; delete config;
delete serialized_engine; delete serialized_engine;
delete builder;
} }
void deserialize_engine(std::string& engine_name, IRuntime** runtime, ICudaEngine** engine, IExecutionContext** context) { void deserialize_engine(std::string& engine_name, IRuntime** runtime, ICudaEngine** engine, IExecutionContext** context) {