add parameter with_pool

PaddlePaddle · Nov 2, 2021 · 069192d · 069192d
1 parent c375848
commit 069192d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/docs/api/paddle/vision/models/ShuffleNetV2_cn.rst b/docs/api/paddle/vision/models/ShuffleNetV2_cn.rst
@@ -3,15 +3,16 @@
 ShuffleNetV2
 -------------------------------
 
-.. py:class:: paddle.vision.models.ShuffleNetV2(num_classes=1000, scale=1.0, act="relu")
+.. py:class:: paddle.vision.models.ShuffleNetV2(scale=1.0, act="relu", num_classes=1000, with_pool=True)
 
  ShuffleNetV2模型，来自论文 `"ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design" <https://arxiv.org/pdf/1807.11164.pdf>`_ 。
 
 参数
 :::::::::
-  - **num_classes** (int, 可选) - 最后一个全连接层输出的维度。如果该值小于0，则不定义最后一个全连接层。默认值：1000。
   - **scale** (float，可选) - 模型通道数的缩放比例。默认值：1.0。
   - **act** (str, 可选) - 网络中使用的激活函数。默认值："relu"。
+  - **num_classes** (int, 可选) - 最后一个全连接层输出的维度。如果该值小于0，则不定义最后一个全连接层。默认值：1000。
+  - **with_pool** (bool，可选) - 是否定义最后一个全连接层之前的池化层。默认值：True。
 
 返回
 :::::::::