浏览代码

修复打印时间bug

leon 5 月之前
父节点
当前提交
ec0a67a0d0
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      src/main.cpp

+ 1 - 0
src/main.cpp

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