tensorRT serialization OK

This commit is contained in:
Francesco Gatti
2017-08-11 17:17:05 +02:00
parent 57c9a6ec99
commit 3b2f062dd9
7 changed files with 54 additions and 14 deletions
+1 -1
View File
@@ -60,7 +60,7 @@ int main() {
net.print();
//convert network to tensorRT
tkDNN::NetworkRT netRT(&net);
tkDNN::NetworkRT netRT(&net, "yolo-tiny.rt");
dnnType *out_data, *out_data2; // cudnn output, tensorRT output