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

View File

@ -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()
m.eval()