mirrors
/
mmclassification
mirror of https://github.com/open-mmlab/mmclassification.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix linting

pull/2/head
lixiaojie 2020-06-11 16:01:54 +08:00
parent 72bbfbc489
commit 93612db881
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mmcls/models/backbones/shufflenet_v1.py
View File

@ -3,7 +3,7 @@ import logging
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.utils.checkpoint as cp 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) constant_init, kaiming_init)
from mmcv.runner import load_checkpoint from mmcv.runner import load_checkpoint
from torch.nn.modules.batchnorm import _BatchNorm from torch.nn.modules.batchnorm import _BatchNorm
@ -185,7 +185,6 @@ class ShuffleUnit(nn.Module):
else: else:
out = _inner_forward(x) out = _inner_forward(x)
return out return out
@ -273,8 +272,8 @@ class ShuffleNetv1(BaseBackbone):
self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, padding=1) self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, padding=1)
self.layer1 = self._make_layer(channels[0], blocks[0], self.layer1 = self._make_layer(
first_block=False) channels[0], blocks[0], first_block=False)
self.layer2 = self._make_layer(channels[1], blocks[1]) self.layer2 = self._make_layer(channels[1], blocks[1])
self.layer3 = self._make_layer(channels[2], blocks[2]) self.layer3 = self._make_layer(channels[2], blocks[2])
@ -361,4 +360,4 @@ class ShuffleNetv1(BaseBackbone):
if mode and self.norm_eval: if mode and self.norm_eval:
for m in self.modules(): for m in self.modules():
if isinstance(m, _BatchNorm): if isinstance(m, _BatchNorm):
m.eval() m.eval()