tensorRT serialization OK
This commit is contained in:
+1
-1
@@ -108,7 +108,7 @@ int main() {
|
||||
net.print();
|
||||
|
||||
//convert network to tensorRT
|
||||
tkDNN::NetworkRT netRT(&net);
|
||||
tkDNN::NetworkRT netRT(&net, "yolo.rt");
|
||||
|
||||
dnnType *out_data, *out_data2; // cudnn output, tensorRT output
|
||||
|
||||
|
||||
Reference in New Issue
Block a user