File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
neural_compressor/torch/quantization Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -905,7 +905,7 @@ def __init__(
905
905
enable_minmax_tuning : bool = True ,
906
906
lr : float = None ,
907
907
minmax_lr : float = None ,
908
- low_gpu_mem_usage : bool = True ,
908
+ low_gpu_mem_usage : bool = False ,
909
909
iters : int = 200 ,
910
910
seqlen : int = 2048 ,
911
911
nsamples : int = 128 ,
@@ -938,7 +938,7 @@ def __init__(
938
938
enable_minmax_tuning (bool): Whether to enable min-max tuning (default is True).
939
939
lr (float): The learning rate (default is 0.005).
940
940
minmax_lr (float): The learning rate for min-max tuning (default is None).
941
- low_gpu_mem_usage (bool): Whether to use low GPU memory (default is True ).
941
+ low_gpu_mem_usage (bool): Whether to use low GPU memory (default is False ).
942
942
iters (int): Number of iterations (default is 200).
943
943
seqlen (int): Length of the sequence.
944
944
nsamples (int): Number of samples (default is 512).
You can’t perform that action at this time.
0 commit comments