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
@@ -22,7 +22,7 @@ int main() {
tkDNN::Dense l6(&net, 10, d3_bin);
tkDNN::Softmax l7(&net);
tkDNN::NetworkRT netRT(&net);
tkDNN::NetworkRT netRT(&net, "mnist.rt");
// Load input
dnnType *data;