From 940c955523d168aed0f908e5f9060ee4939bb11f Mon Sep 17 00:00:00 2001 From: lixiaojie Date: Fri, 12 Jun 2020 20:27:07 +0800 Subject: [PATCH] fix description --- mmcls/models/backbones/shufflenet_v1.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/mmcls/models/backbones/shufflenet_v1.py b/mmcls/models/backbones/shufflenet_v1.py index 15923134..f834bde3 100644 --- a/mmcls/models/backbones/shufflenet_v1.py +++ b/mmcls/models/backbones/shufflenet_v1.py @@ -86,7 +86,7 @@ class ShuffleUnit(nn.Module): will save some memory while slowing down the training speed. Returns: - Tensor: output tensor + Tensor: The output tensor. """ def __init__(self, @@ -291,17 +291,16 @@ class ShuffleNetv1(BaseBackbone): raise TypeError('pretrained must be a str or None') def _make_layer(self, outplanes, blocks, first_block=True): - """ Stack n bottleneck modules where n is inferred from the depth of - the network. + """ Stack ShuffleUnit blocks to make a layer. Args: - outplanes: number of output channels - blocks: number of blocks to be built + outplanes: Number of output channels. + blocks: Number of blocks to be built. first_block (bool, optional): Whether is the first ShuffleUnit of a sequential ShuffleUnits. If True, use the grouped 1x1 convolution. Returns: - Module: a module consisting of n sequential ShuffleUnits. + Module: A module consisting of several ShuffleUnit blocks. """ layers = []