Unverified Commit cac8a97b authored by Zhiqiang Wang's avatar Zhiqiang Wang Committed by GitHub
Browse files

Keep consistency in Classes ConvBNActivation (#3750)



* Keep consistency of ConvBNActivation

* Simplify using the Python3 idiom
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 1b0bd0e3
...@@ -46,7 +46,7 @@ class ConvBNActivation(nn.Sequential): ...@@ -46,7 +46,7 @@ class ConvBNActivation(nn.Sequential):
norm_layer = nn.BatchNorm2d norm_layer = nn.BatchNorm2d
if activation_layer is None: if activation_layer is None:
activation_layer = nn.ReLU6 activation_layer = nn.ReLU6
super(ConvBNReLU, self).__init__( super().__init__(
nn.Conv2d(in_planes, out_planes, kernel_size, stride, padding, dilation=dilation, groups=groups, nn.Conv2d(in_planes, out_planes, kernel_size, stride, padding, dilation=dilation, groups=groups,
bias=False), bias=False),
norm_layer(out_planes), norm_layer(out_planes),
......
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