fix benchmark_model.cc

AI-performance · Sep 6, 2020 · 52423da · 52423da
1 parent 1b2cd0d
commit 52423da
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tflite/benchmark_model.cc b/tflite/benchmark_model.cc
@@ -52,10 +52,10 @@ void BenchmarkLoggingListener::OnBenchmarkEnd(const BenchmarkResults& results) {
   auto init_mem_usage = results.init_mem_usage();
   auto overall_mem_usage = results.overall_mem_usage();
 
-  std::cout << "time cost: min=" << inference_us.min() / 1000.
-            << " max=" << inference_us.max() / 1000.
-            << " avg=" << inference_us.avg() / 1000.
-            << " std_dev=" << inference_us.std_deviation() / 1000.
+  std::cout << "time cost: min=" << inference_us.min() / 1000. << "ms"
+            << " max=" << inference_us.max() / 1000. << "ms"
+            << " avg=" << inference_us.avg() / 1000. << "ms"
+            << " std_dev=" << inference_us.std_deviation() / 1000. << "ms"
             << std::endl;
   TFLITE_LOG(INFO) << "Inference timings in us: "
                    << "Init: " << init_us << ", "