From 405bb94bd1ef51b51c9fe67ade00c262b169e3ce Mon Sep 17 00:00:00 2001 From: Wojciech Uss Date: Thu, 20 Feb 2020 05:01:19 +0100 Subject: [PATCH] Fix batch_num default value (#22682) --- .../slim/tests/qat_int8_image_classification_comparison.py | 5 +++-- .../fluid/contrib/slim/tests/qat_int8_nlp_comparison.py | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) 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 0a0359dc6e3..24ec7c79769 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 b6eb825b09d..d51db0db084 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, -- GitLab