leon 5 сар өмнө
parent
commit
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");
   }
 }