Skip to content

Commit

Permalink
fix doc style (#61688)
Browse files Browse the repository at this point in the history
  • Loading branch information
GreatV committed Feb 19, 2024
1 parent e5a85b6 commit 12e5c97
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions python/paddle/quantization/quantize.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,8 +28,9 @@
class Quantization(metaclass=abc.ABCMeta):
r"""
Abstract class used to prepares a copy of the model for quantization calibration or quantization-aware training.
Args:
config(QuantConfig) - Quantization configuration
config(QuantConfig): Quantization configuration
"""

def __init__(self, config: QuantConfig):
Expand All @@ -43,10 +44,11 @@ def quantize(self, model: Layer, inplace=False):
def convert(self, model: Layer, inplace=False, remain_weight=False):
r"""Convert the quantization model to ONNX style. And the converted
model can be saved as inference model by calling paddle.jit.save.
Args:
model(Layer) - The quantized model to be converted.
inplace(bool, optional) - Whether to modify the model in-place, default is False.
remain_weight(bool, optional) - Whether to remain weights in floats, default is False.
model(Layer): The quantized model to be converted.
inplace(bool, optional): Whether to modify the model in-place, default is False.
remain_weight(bool, optional): Whether to remain weights in floats, default is False.
Return: The converted model
Expand Down

0 comments on commit 12e5c97

Please sign in to comment.