Commit 4b2abfcf authored by Kai Chen's avatar Kai Chen
Browse files

minor format adjustment

parent 64afdd0a
...@@ -4,7 +4,6 @@ from .resnet import ResNet, make_res_layer ...@@ -4,7 +4,6 @@ from .resnet import ResNet, make_res_layer
from .weight_init import xavier_init, normal_init, uniform_init, kaiming_init from .weight_init import xavier_init, normal_init, uniform_init, kaiming_init
__all__ = [ __all__ = [
'AlexNet', 'VGG', 'make_vgg_layer', 'AlexNet', 'VGG', 'make_vgg_layer', 'ResNet', 'make_res_layer',
'ResNet', 'make_res_layer', 'xavier_init', 'normal_init', 'uniform_init', 'xavier_init', 'normal_init', 'uniform_init', 'kaiming_init'
'kaiming_init'
] ]
...@@ -16,15 +16,10 @@ def conv3x3(in_planes, out_planes, dilation=1, bias=False): ...@@ -16,15 +16,10 @@ def conv3x3(in_planes, out_planes, dilation=1, bias=False):
bias=bias) bias=bias)
def make_vgg_layer(inplanes, def make_vgg_layer(inplanes, planes, num_blocks, dilation=1, with_bn=False):
planes,
num_blocks,
dilation=1,
with_bn=False):
layers = [] layers = []
for _ in range(num_blocks): for _ in range(num_blocks):
layers.append( layers.append(conv3x3(inplanes, planes, dilation, not with_bn))
conv3x3(inplanes, planes, dilation, not with_bn))
if with_bn: if with_bn:
layers.append(nn.BatchNorm2d(planes)) layers.append(nn.BatchNorm2d(planes))
layers.append(nn.ReLU(inplace=True)) layers.append(nn.ReLU(inplace=True))
...@@ -118,9 +113,7 @@ class VGG(nn.Module): ...@@ -118,9 +113,7 @@ class VGG(nn.Module):
for m in self.modules(): for m in self.modules():
if isinstance(m, nn.Conv2d): if isinstance(m, nn.Conv2d):
nn.init.kaiming_normal_( nn.init.kaiming_normal_(
m.weight, m.weight, mode='fan_out', nonlinearity='relu')
mode='fan_out',
nonlinearity='relu')
if m.bias is not None: if m.bias is not None:
nn.init.constant_(m.bias, 0) nn.init.constant_(m.bias, 0)
elif isinstance(m, nn.BatchNorm2d): elif isinstance(m, nn.BatchNorm2d):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment