inference fix

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;