Skip to content

Commit caf3002

Browse files
authored
fix quant_post batch_size (PaddlePaddle#1006)
1 parent fc090b2 commit caf3002

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddleslim/quant/quanter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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.

0 commit comments

Comments
 (0)