mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

Tensorrt8 #285

Merged
ceccocats merged 71 commits from tensorrt8 into master 2022-03-30 17:45:41 +02:00
Conversation 0 Commits 71 Files Changed 126
+1 -5
2 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 504dee5016 - Show all commits
tests/darknet/yolo4.cpp
+1
View File
@@ -29,6 +29,7 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT); int ret = testInference(input_bins, output_bins, net, netRT);
net->releaseLayers(); net->releaseLayers();
delete net; delete net;
netRT->destroy();
delete netRT; delete netRT;
return ret; return ret;
} }
tests/darknet/yolo4tiny.cpp
-5
View File
@@ -27,14 +27,9 @@ int main() {
int ret = testInference(input_bins, output_bins, net, netRT); int ret = testInference(input_bins, output_bins, net, netRT);
std::cout<<ret<<std::endl; std::cout<<ret<<std::endl;
std::cout<<"Releasing Layers"<<std::endl;
net->releaseLayers(); net->releaseLayers();
std::cout<<"DELETING NET"<<std::endl;
delete net; delete net;
std::cout<<"DELETING NETRT"<<std::endl;
netRT->destroy(); netRT->destroy();
delete netRT; delete netRT;
std::cout<<"RETRUNING RET"<<std::endl;
return ret; return ret;
} }