diff --git a/deploy/slim/quantization/quant_kl.py b/deploy/slim/quantization/quant_kl.py
index 83bd671456aa4b026acde8c8e23af5490bb7fcbf..d866784ae6a3c087215320ec95bd39fdd1e89418 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__':