Skip to content

Commit 553099a

Browse files
authored
update ResNet doc
1 parent bf4aa3b commit 553099a

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

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

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

6-
.. py:class:: paddle.vision.models.ResNet(Block, depth=50, num_classes=1000, with_pool=True)
6+
.. py:class:: paddle.vision.models.ResNet(Block, depth=50, width=64, num_classes=1000, with_pool=True)
77
88
ResNet模型,来自论文 `"Deep Residual Learning for Image Recognition" <https://arxiv.org/pdf/1512.03385.pdf>`_ 。
99

1010
参数
1111
:::::::::
1212
- **Block** (BasicBlock|BottleneckBlock) - 模型的残差模块。
13-
- **depth** (int,可选) - resnet模型的深度。默认值:50
13+
- **depth** (int,可选) - resnet模型的深度。默认值:50。
14+
- **width** (int,可选) - resnet模型的基础宽度。默认值:64。
1415
- **num_classes** (int, 可选) - 最后一个全连接层输出的维度。如果该值小于0,则不定义最后一个全连接层。默认值:1000。
1516
- **with_pool** (bool,可选) - 是否定义最后一个全连接层之前的池化层。默认值:True。
1617

@@ -28,6 +29,8 @@ ResNet模型,Layer的实例。
2829
2930
resnet50 = ResNet(BottleneckBlock, 50)
3031
32+
wide_resnet50_2 = ResNet(BottleneckBlock, 50, width=64*2)
33+
3134
resnet18 = ResNet(BasicBlock, 18)
3235
3336
x = paddle.rand([1, 3, 224, 224])

0 commit comments

Comments
 (0)