提交 5a2fc5b5 编写于 作者: T tensor-tang

fix print error

上级 3ea19b75
...@@ -182,6 +182,7 @@ void TestLACPrediction(const std::string &model_path, ...@@ -182,6 +182,7 @@ void TestLACPrediction(const std::string &model_path,
Timer timer; Timer timer;
if (test_all_data) { if (test_all_data) {
double sum = 0; double sum = 0;
LOG(INFO) << "Total number of samples: " << data.datasets.size();
for (int i = 0; i < repeat; i++) { for (int i = 0; i < repeat; i++) {
for (size_t bid = 0; bid < data.batched_datas.size(); ++bid) { for (size_t bid = 0; bid < data.batched_datas.size(); ++bid) {
GetOneBatch(&input_slots, &data, batch_size); GetOneBatch(&input_slots, &data, batch_size);
...@@ -190,7 +191,9 @@ void TestLACPrediction(const std::string &model_path, ...@@ -190,7 +191,9 @@ void TestLACPrediction(const std::string &model_path,
sum += timer.toc(); sum += timer.toc();
} }
} }
PrintTime(batch_size, repeat, 1, 0, sum / batch_size); PrintTime(batch_size, repeat, 1, 0, sum / repeat);
LOG(INFO) << "Average latency of each sample: "
<< sum / repeat / data.datasets.size() << " ms";
return; return;
} }
timer.tic(); timer.tic();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册