File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -312,7 +312,7 @@ def quant_post_static(
312312 params_filename = None ,
313313 save_model_filename = '__model__' ,
314314 save_params_filename = '__params__' ,
315- batch_size = 16 ,
315+ batch_size = 1 ,
316316 batch_nums = None ,
317317 scope = None ,
318318 algo = 'hist' ,
@@ -359,7 +359,7 @@ def quant_post_static(
359359 save_model_filename(str): The name of model file to save the quantized inference program. Default: '__model__'.
360360 save_params_filename(str): The name of file to save all related parameters.
361361 If it is set None, parameters will be saved in separate files. Default: '__params__'.
362- batch_size(int, optional): The batch size of DataLoader, default is 16 .
362+ batch_size(int, optional): The batch size of DataLoader, default is 1 .
363363 batch_nums(int, optional): If batch_nums is not None, the number of calibrate
364364 data is 'batch_size*batch_nums'. If batch_nums is None, use all data
365365 generated by sample_generator as calibrate data.
You can’t perform that action at this time.
0 commit comments