set batch_size=1 in kl quant

2d124b74 · LDOUBLEV · d57049bc · 2d124b74
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

deploy/slim/quantization/quant_kl.py deploy/slim/quantization/quant_kl.py +2 -1

未找到文件。
--- 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__':