From 2d124b74ed0de6c3ed7022ffa93ab721ed83443f Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Thu, 9 Sep 2021 12:21:59 +0000
Subject: [PATCH] set batch_size=1 in kl quant

---
 deploy/slim/quantization/quant_kl.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/deploy/slim/quantization/quant_kl.py b/deploy/slim/quantization/quant_kl.py
index 83bd6714..d866784a 100755
--- a/deploy/slim/quantization/quant_kl.py
+++ b/deploy/slim/quantization/quant_kl.py
@@ -137,7 +137,8 @@ def main(config, device, logger, vdl_writer):
         sample_generator=sample_generator(train_dataloader),
         save_model_filename='inference.pdmodel',
         save_params_filename='inference.pdiparams',
-        batch_nums=10)
+        batch_size=1,
+        batch_nums=None)
 
 
 if __name__ == '__main__':
-- 
GitLab