Commit 00dc3fe8 authored by Federico Rossi's avatar Federico Rossi

fast version

parent fe42bcd0
......@@ -19,8 +19,8 @@ int main(int argc, char **argv) {
std::vector<tiny_dnn::vec_t> vimg;
std::vector<unsigned int> times;
parse_binImagenet<32,32,3,uint8_t>(data_path,&vimg,&vlab,-1,1);
int successes = 0;
net.test(vimg, vlab).print_detail(std::cout);
/*int successes = 0;
auto start = get_time::now();
for(unsigned int i = 0; i < 1000; ++i) {
......@@ -41,5 +41,5 @@ int main(int argc, char **argv) {
make_test_report(vimg,successes,times,std::chrono::duration_cast<ms>(diff).count(),"../../plot/test_perf_");
std::cout << "Accuracy: " << successes << "/" << vimg.size() << "\n";
std::cout << "Time elapsed: " << std::chrono::duration_cast<ms>(diff).count() << "ms\n";
return 0;
return 0;*/
}
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment