From 93612db881b4b51ea2b2fd6f5bf9bede1982d9bc Mon Sep 17 00:00:00 2001 From: lixiaojie Date: Thu, 11 Jun 2020 16:01:54 +0800 Subject: [PATCH] fix linting --- mmcls/models/backbones/shufflenet_v1.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/mmcls/models/backbones/shufflenet_v1.py b/mmcls/models/backbones/shufflenet_v1.py index a0d66129c..0499bfc9c 100644 --- a/mmcls/models/backbones/shufflenet_v1.py +++ b/mmcls/models/backbones/shufflenet_v1.py @@ -3,7 +3,7 @@ import logging import torch import torch.nn as nn import torch.utils.checkpoint as cp -from mmcv.cnn import (ConvModule, build_conv_layer, build_activation_layer, +from mmcv.cnn import (ConvModule, build_activation_layer, build_conv_layer, constant_init, kaiming_init) from mmcv.runner import load_checkpoint from torch.nn.modules.batchnorm import _BatchNorm @@ -185,7 +185,6 @@ class ShuffleUnit(nn.Module): else: out = _inner_forward(x) - return out @@ -273,8 +272,8 @@ class ShuffleNetv1(BaseBackbone): self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, padding=1) - self.layer1 = self._make_layer(channels[0], blocks[0], - first_block=False) + self.layer1 = self._make_layer( + channels[0], blocks[0], first_block=False) self.layer2 = self._make_layer(channels[1], blocks[1]) self.layer3 = self._make_layer(channels[2], blocks[2]) @@ -361,4 +360,4 @@ class ShuffleNetv1(BaseBackbone): if mode and self.norm_eval: for m in self.modules(): if isinstance(m, _BatchNorm): - m.eval() \ No newline at end of file + m.eval()