diff --git a/python/paddle/fluid/contrib/slim/tests/qat_int8_image_classification_comparison.py b/python/paddle/fluid/contrib/slim/tests/qat_int8_image_classification_comparison.py index 0a0359dc6e32ca996db3385a3098795c14b2706d..24ec7c79769df98e79fd0d712e636921bc9b9d34 100644 --- a/python/paddle/fluid/contrib/slim/tests/qat_int8_image_classification_comparison.py +++ b/python/paddle/fluid/contrib/slim/tests/qat_int8_image_classification_comparison.py @@ -57,8 +57,9 @@ def parse_args(): parser.add_argument( '--batch_num', type=int, - default=1, - help='Number of batches to process. 0 or less means all.') + default=0, + help='Number of batches to process. 0 or less means whole dataset. Default: 0.' + ) parser.add_argument( '--acc_diff_threshold', type=float, diff --git a/python/paddle/fluid/contrib/slim/tests/qat_int8_nlp_comparison.py b/python/paddle/fluid/contrib/slim/tests/qat_int8_nlp_comparison.py index b6eb825b09d2c0aad0d45049feafc09106d1e8c7..d51db0db08472450e824b08f73363cb6734198c3 100644 --- a/python/paddle/fluid/contrib/slim/tests/qat_int8_nlp_comparison.py +++ b/python/paddle/fluid/contrib/slim/tests/qat_int8_nlp_comparison.py @@ -57,8 +57,9 @@ def parse_args(): parser.add_argument( '--batch_num', type=int, - default=1, - help='Number of batches to process. 0 or less means all.') + default=0, + help='Number of batches to process. 0 or less means whole dataset. Default: 0.' + ) parser.add_argument( '--acc_diff_threshold', type=float,