Skip to content

Commit 069192d

Browse files
authored
add parameter with_pool
1 parent c375848 commit 069192d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

docs/api/paddle/vision/models/ShuffleNetV2_cn.rst

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,15 +3,16 @@
33
ShuffleNetV2
44
-------------------------------
55

6-
.. py:class:: paddle.vision.models.ShuffleNetV2(num_classes=1000, scale=1.0, act="relu")
6+
.. py:class:: paddle.vision.models.ShuffleNetV2(scale=1.0, act="relu", num_classes=1000, with_pool=True)
77
88
ShuffleNetV2模型,来自论文 `"ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design" <https://arxiv.org/pdf/1807.11164.pdf>`_ 。
99

1010
参数
1111
:::::::::
12-
- **num_classes** (int, 可选) - 最后一个全连接层输出的维度。如果该值小于0,则不定义最后一个全连接层。默认值:1000。
1312
- **scale** (float,可选) - 模型通道数的缩放比例。默认值:1.0。
1413
- **act** (str, 可选) - 网络中使用的激活函数。默认值:"relu"。
14+
- **num_classes** (int, 可选) - 最后一个全连接层输出的维度。如果该值小于0,则不定义最后一个全连接层。默认值:1000。
15+
- **with_pool** (bool,可选) - 是否定义最后一个全连接层之前的池化层。默认值:True。
1516

1617
返回
1718
:::::::::

0 commit comments

Comments
 (0)