@@ -42,24 +42,21 @@ std::chrono::milliseconds measure_time_tbb(
4242// Function: main
4343int main (int argc, char *argv[]){
4444
45- CLI::App app{" MNIST" };
45+ CLI::App app{" MNIST" };
4646
4747 unsigned num_threads {1 };
48- app.add_option (" -t,--num_threads" , num_threads, " number of threads (default=1)" );
48+ app.add_option (" -t,--num_threads" , num_threads, " number of threads (default=1)" );
4949
5050 unsigned num_epochs {10 };
51- app.add_option (" -e,--num_epochs" , num_epochs, " number of epochs (default=10)" );
52-
53- unsigned num_rounds {1 };
54- app.add_option (" -r,--num_rounds" , num_rounds, " number of rounds (default=1)" );
55-
56- std::string model = " tf" ;
57- app.add_option (" -m,--model" , model, " model name (tbb|omp|tf(default))" );
58-
59- CLI11_PARSE (app, argc, argv);
51+ app.add_option (" -e,--num_epochs" , num_epochs, " number of epochs (default=10)" );
6052
53+ unsigned num_rounds {1 };
54+ app.add_option (" -r,--num_rounds" , num_rounds, " number of rounds (default=1)" );
6155
56+ std::string model = " tf" ;
57+ app.add_option (" -m,--model" , model, " model name (tbb|omp|tf(default))" );
6258
59+ CLI11_PARSE (app, argc, argv);
6360
6461 double runtime {0.0 };
6562
0 commit comments