diff --git a/demo/demo/demo.cpp b/demo/demo/demo.cpp index 67d5786..afa5577 100644 --- a/demo/demo/demo.cpp +++ b/demo/demo/demo.cpp @@ -131,7 +131,7 @@ int main(int argc, char *argv[]) { std::cout<<"Min: "<<*std::min_element(detNN->stats.begin(), detNN->stats.end())/n_batch<<" ms\n"; std::cout<<"Max: "<<*std::max_element(detNN->stats.begin(), detNN->stats.end())/n_batch<<" ms\n"; for(int i=0; istats.size(); i++) mean += detNN->stats[i]; mean /= detNN->stats.size(); - std::cout<<"Avg: "<input_dim; dim.n = nBatches; { - dim.print(); + if(VERBOSE) dim.print(); TIMER_START netRT->infer(dim, input_d); TIMER_STOP - dim.print(); + if(VERBOSE) dim.print(); stats.push_back(t_ns); if(save_times) *times<