From 80d3a20bf1a4b5df1dff251a3fbefa75d9d58f3c Mon Sep 17 00:00:00 2001
From: Candy2Tang <141831089+Candy2Tang@users.noreply.github.com>
Date: Mon, 21 Aug 2023 12:01:39 +0800
Subject: [PATCH] [xdoctest][task 122] Reformat example code with google style
 in python/paddle/quantization/qat.py (#56233)

* [xdoctest][task 122] test=docs_preview

* test=document_fix

* fix indent

---------

Co-authored-by: SigureMo <sigure.qaq@gmail.com>
---
 python/paddle/quantization/qat.py | 74 ++++++++++++++++++++++++-------
 1 file changed, 59 insertions(+), 15 deletions(-)

diff --git a/python/paddle/quantization/qat.py b/python/paddle/quantization/qat.py
index cff2304acbe..b851e473b60 100644
--- a/python/paddle/quantization/qat.py
+++ b/python/paddle/quantization/qat.py
@@ -28,11 +28,12 @@ class QAT(Quantization):
 
     Examples:
         .. code-block:: python
-            from paddle.quantization import QAT, QuantConfig
-            from paddle.quantization.quanters import FakeQuanterWithAbsMaxObserver
-            quanter = FakeQuanterWithAbsMaxObserver(moving_rate=0.9)
-            q_config = QuantConfig(activation=quanter, weight=quanter)
-            qat = QAT(q_config)
+
+            >>> from paddle.quantization import QAT, QuantConfig
+            >>> from paddle.quantization.quanters import FakeQuanterWithAbsMaxObserver
+            >>> quanter = FakeQuanterWithAbsMaxObserver(moving_rate=0.9)
+            >>> q_config = QuantConfig(activation=quanter, weight=quanter)
+            >>> qat = QAT(q_config)
     """
 
     def __init__(self, config: QuantConfig):
@@ -52,17 +53,60 @@ class QAT(Quantization):
         Return: The prepared model for quantization-aware training.
 
         Examples:
-        .. code-block:: python
-            from paddle.quantization import QAT, QuantConfig
-            from paddle.quantization.quanters import FakeQuanterWithAbsMaxObserver
-            from paddle.vision.models import LeNet
+            .. code-block:: python
+
+                >>> from paddle.quantization import QAT, QuantConfig
+                >>> from paddle.quantization.quanters import FakeQuanterWithAbsMaxObserver
+                >>> from paddle.vision.models import LeNet
 
-            quanter = FakeQuanterWithAbsMaxObserver(moving_rate=0.9)
-            q_config = QuantConfig(activation=quanter, weight=quanter)
-            qat = QAT(q_config)
-            model = LeNet()
-            quant_model = qat.quantize(model)
-            print(quant_model)
+                >>> quanter = FakeQuanterWithAbsMaxObserver(moving_rate=0.9)
+                >>> q_config = QuantConfig(activation=quanter, weight=quanter)
+                >>> qat = QAT(q_config)
+                >>> model = LeNet()
+                >>> quant_model = qat.quantize(model)
+                >>> print(quant_model)
+                LeNet(
+                  (features): Sequential(
+                    (0): QuantedConv2D(
+                      (weight_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                      (activation_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                    )
+                    (1): ObserveWrapper(
+                      (_observer): FakeQuanterWithAbsMaxObserverLayer()
+                      (_observed): ReLU()
+                    )
+                    (2): ObserveWrapper(
+                      (_observer): FakeQuanterWithAbsMaxObserverLayer()
+                      (_observed): MaxPool2D(kernel_size=2, stride=2, padding=0)
+                    )
+                    (3): QuantedConv2D(
+                      (weight_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                      (activation_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                    )
+                    (4): ObserveWrapper(
+                      (_observer): FakeQuanterWithAbsMaxObserverLayer()
+                      (_observed): ReLU()
+                    )
+                    (5): ObserveWrapper(
+                      (_observer): FakeQuanterWithAbsMaxObserverLayer()
+                      (_observed): MaxPool2D(kernel_size=2, stride=2, padding=0)
+                    )
+                  )
+                  (fc): Sequential(
+                    (0): QuantedLinear(
+                      (weight_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                      (activation_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                    )
+                    (1): QuantedLinear(
+                      (weight_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                      (activation_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                    )
+                    (2): QuantedLinear(
+                      (weight_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                      (activation_quanter): FakeQuanterWithAbsMaxObserverLayer()
+                    )
+                  )
+                )
         """
         assert (
             model.training
-- 
GitLab