From 461e07d65dc3b7ad0e38f52c6a263f10bc72697b Mon Sep 17 00:00:00 2001 From: lixiaojie Date: Sun, 14 Jun 2020 03:51:42 +0800 Subject: [PATCH] Update default value --- mmcls/models/backbones/shufflenet_v2.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/mmcls/models/backbones/shufflenet_v2.py b/mmcls/models/backbones/shufflenet_v2.py index 4e5867ff..da092fe1 100644 --- a/mmcls/models/backbones/shufflenet_v2.py +++ b/mmcls/models/backbones/shufflenet_v2.py @@ -65,13 +65,17 @@ class InvertedResidual(nn.Module): inplanes (int): The input channels of the block. planes (int): The output channels of the block. stride (int): stride of the 3x3 convolution layer. Default: 1 - conv_cfg (dict): Config dict for convolution layer. Default: None, - which means using conv2d. - norm_cfg (dict): Config dict for normalization layer. Default: None. + conv_cfg (dict): Config dict for convolution layer. + Default: None, which means using conv2d. + norm_cfg (dict): Config dict for normalization layer. + Default: dict(type='BN'). act_cfg (dict): Config dict for activation layer. Default: dict(type='ReLU'). - with_cp (bool, optional): Use checkpoint or not. Using checkpoint - will save some memory while slowing down the training speed. + norm_eval (bool): Whether to set norm layers to eval mode, namely, + freeze running stats (mean and var). Note: Effect on Batch Norm + and its variants only. Default: True. + with_cp (bool): Use checkpoint or not. Using checkpoint will save some + memory while slowing down the training speed. Default: False. Returns: Tensor: The output tensor.