Updated the MLP test to accept the number of epochs. Reduced the epoc…

…hs in ci_test.sh to shorten the CI build time (apache#14149)
haohuanw · Jun 23, 2019 · 788f821 · 788f821
1 parent 646f4d5
commit 788f821
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/cpp-package/example/mlp.cpp b/cpp-package/example/mlp.cpp
@@ -49,7 +49,7 @@ void OutputAccuracy(mx_float* pred, mx_float* target) {
   std::cout << "Accuracy: " << right / 128.0 << std::endl;
 }
 
-void MLP() {
+void MLP(int max_epoch) {
   auto sym_x = Symbol::Variable("X");
   auto sym_label = Symbol::Variable("label");
 
@@ -144,7 +144,6 @@ void MLP() {
                                grad_req_type, aux_states);
 
   std::cout << "Training" << std::endl;
-  int max_epoch = 15000;
   mx_float learning_rate = 0.0001;
   for (int epoch_num = 0; epoch_num < max_epoch; ++epoch_num) {
     exe->Forward(true);
@@ -173,8 +172,8 @@ void MLP() {
 }
 
 int main(int argc, char** argv) {
-  MLP();
+  int max_epoch = argc > 1 ? strtol(argv[1], NULL, 10) : 15000;
+  MLP(max_epoch);
   MXNotifyShutdown();
   return 0;
 }
-
diff --git a/cpp-package/tests/ci_test.sh b/cpp-package/tests/ci_test.sh
@@ -25,22 +25,22 @@ ls -l ../../lib/
 ./get_data.sh
 
 cp ../../build/cpp-package/example/lenet .
-./lenet 10
+./lenet 1
 
 cp ../../build/cpp-package/example/alexnet .
 ./alexnet 1
 
 cp ../../build/cpp-package/example/lenet_with_mxdataiter .
-./lenet_with_mxdataiter 5
+./lenet_with_mxdataiter 1
 
 cp ../../build/cpp-package/example/resnet .
-./resnet 5
+./resnet 1
 
 cp ../../build/cpp-package/example/inception_bn .
-./inception_bn 5
+./inception_bn 1
 
 cp ../../build/cpp-package/example/mlp .
-./mlp
+./mlp 150
 
 cp ../../build/cpp-package/example/mlp_cpu .
 ./mlp_cpu