From 6110fffbb5efbb6d27672aaa05388dcf9279e0b8 Mon Sep 17 00:00:00 2001
From: Francesco Gatti <gattifrancesco@hotmail.it>
Date: Mon, 14 Aug 2017 11:36:48 +0200
Subject: [PATCH] inference fix

---
 tests/test_rtinference/rtinference.cpp | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/tests/test_rtinference/rtinference.cpp b/tests/test_rtinference/rtinference.cpp
index 3c61083..a731394 100644
--- a/tests/test_rtinference/rtinference.cpp
+++ b/tests/test_rtinference/rtinference.cpp
@@ -3,23 +3,22 @@
 
 int main(int argc, char *argv[]) {
 
-    // Network layout
-    tkDNN::dataDim_t dim(1, 3, 608, 608, 1);
-    tkDNN::Network net(dim);
-
     if(argc < 2 || !fileExist(argv[1]))
         FatalError("unable to read serialRT file");
 
     //convert network to tensorRT
-    tkDNN::NetworkRT netRT(&net, argv[1]);
+    tkDNN::NetworkRT netRT(NULL, argv[1]);
 
+    tkDNN::dataDim_t dim = netRT.input_dim;
     dnnType *data;
     checkCuda(cudaMalloc(&data, dim.tot()*sizeof(dnnType)));
 
     printCenteredTitle(" TENSORRT inference ", '=', 30); {
+        dim.print();
         TIMER_START
         data = netRT.infer(dim, data);
         TIMER_STOP
+        dim.print();
     }
     
     return 0;