diff --git a/paddle/trainer/Trainer.cpp b/paddle/trainer/Trainer.cpp
index 7fc48dd1fbec6588b71db031d89dd88c5c5cf92c..32c4bad239ec901862a94e5996e171dc5be3cdc3 100644
--- a/paddle/trainer/Trainer.cpp
+++ b/paddle/trainer/Trainer.cpp
@@ -46,6 +46,18 @@ P_DEFINE_int32(test_period, 0,
                " If not 0, test log_period batches."
                " If 0, test on all test data");
 
+P_DEFINE_int32(test_batches_while_training, 0,
+               "Run test every so many train batches."
+               " 0 for testing after each pass."
+               " If not 0, test log_period batches."
+               " If 0, test on all test data");
+
+P_DEFINE_int32(test_batches_while_end, 0,
+               "Run test every so many train batches."
+               " 0 for testing after each pass."
+               " If not 0, test log_period batches."
+               " If 0, test on all test data");
+
 P_DEFINE_bool(local, true, "Train in local mode or not");
 
 P_DEFINE_bool(