diff --git a/docs/zh_cn/api_cn/dygraph/quanter/qat.rst b/docs/zh_cn/api_cn/dygraph/quanter/qat.rst
index 3100fd9cc68652cbee3d9e6d8a4a452e385b8e0f..448ecb75d33fb133c657f0d29653e7574e378ee1 100644
--- a/docs/zh_cn/api_cn/dygraph/quanter/qat.rst
+++ b/docs/zh_cn/api_cn/dygraph/quanter/qat.rst
@@ -251,7 +251,8 @@ QAT
     
 
     .. code-block:: python
-
+        
+        import paddle
         from paddleslim import PTQ
         from paddle.vision.models import mobilenet_v1
         fp32_model = mobilenet_v1(pretrained=True)