Skip to content

Commit

Permalink
Revert "fix doc style (PaddlePaddle#61688)"
Browse files Browse the repository at this point in the history
This reverts commit 12e5c97.
  • Loading branch information
hanhaowen-mt committed May 13, 2024
1 parent 3cbfcfc commit f76f3ed
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions python/paddle/quantization/quantize.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,8 @@
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 @@ -44,11 +43,10 @@ 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 f76f3ed

Please sign in to comment.