forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

set batch_size=1 in kl quant

This commit is contained in:
LDOUBLEV 2021-09-09 12:21:59 +00:00
parent d57049bc75
commit 2d124b74ed
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deploy/slim/quantization/quant_kl.py
View File

@ -137,7 +137,8 @@ def main(config, device, logger, vdl_writer):
sample_generator=sample_generator(train_dataloader), sample_generator=sample_generator(train_dataloader),
save_model_filename='inference.pdmodel', save_model_filename='inference.pdmodel',
save_params_filename='inference.pdiparams', save_params_filename='inference.pdiparams',
batch_nums=10) batch_size=1,
batch_nums=None)
if __name__ == '__main__': if __name__ == '__main__':