leon пре 5 месеци
родитељ
комит
ad0197abb2
1 измењених фајлова са 1 додато и 1 уклоњено
  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");
   }
 }