leon 5 mesi fa
parent
commit
ad0197abb2
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      src/main.cpp

+ 1 - 1
src/main.cpp

@@ -18,9 +18,9 @@ void single_inference() {
   trt::Timer timer;
   for (int i = 0; i < 100; i++)
   {
-    trt::Timer timer1;
     auto attr = resnet->forward(cvimg(image));
     printf("score : %lf, label : %d\n", attr.confidence, attr.class_label);
+    timer.stop("BATCH1");
   }
 }