未验证 提交 caf30021 编写于 作者: G Guanghua Yu 提交者: GitHub

fix quant_post batch_size (#1006)

上级 fc090b28
...@@ -312,7 +312,7 @@ def quant_post_static( ...@@ -312,7 +312,7 @@ def quant_post_static(
params_filename=None, params_filename=None,
save_model_filename='__model__', save_model_filename='__model__',
save_params_filename='__params__', save_params_filename='__params__',
batch_size=16, batch_size=1,
batch_nums=None, batch_nums=None,
scope=None, scope=None,
algo='hist', algo='hist',
...@@ -359,7 +359,7 @@ def quant_post_static( ...@@ -359,7 +359,7 @@ def quant_post_static(
save_model_filename(str): The name of model file to save the quantized inference program. Default: '__model__'. save_model_filename(str): The name of model file to save the quantized inference program. Default: '__model__'.
save_params_filename(str): The name of file to save all related parameters. save_params_filename(str): The name of file to save all related parameters.
If it is set None, parameters will be saved in separate files. Default: '__params__'. If it is set None, parameters will be saved in separate files. Default: '__params__'.
batch_size(int, optional): The batch size of DataLoader, default is 16. batch_size(int, optional): The batch size of DataLoader, default is 1.
batch_nums(int, optional): If batch_nums is not None, the number of calibrate batch_nums(int, optional): If batch_nums is not None, the number of calibrate
data is 'batch_size*batch_nums'. If batch_nums is None, use all data data is 'batch_size*batch_nums'. If batch_nums is None, use all data
generated by sample_generator as calibrate data. generated by sample_generator as calibrate data.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册