|
|
@ -797,9 +797,6 @@ class BottleneckBlock(nn.Module):
|
|
|
|
self.shortcut = nn.Identity()
|
|
|
|
self.shortcut = nn.Identity()
|
|
|
|
|
|
|
|
|
|
|
|
self.conv1_1x1 = layers.conv_norm_act(in_chs, mid_chs, 1)
|
|
|
|
self.conv1_1x1 = layers.conv_norm_act(in_chs, mid_chs, 1)
|
|
|
|
self.conv2_kxk = layers.conv_norm_act(
|
|
|
|
|
|
|
|
mid_chs, mid_chs, kernel_size, stride=stride, dilation=dilation[0],
|
|
|
|
|
|
|
|
groups=groups, drop_block=drop_block)
|
|
|
|
|
|
|
|
self.conv2_kxk = layers.conv_norm_act(
|
|
|
|
self.conv2_kxk = layers.conv_norm_act(
|
|
|
|
mid_chs, mid_chs, kernel_size, stride=stride, dilation=dilation[0],
|
|
|
|
mid_chs, mid_chs, kernel_size, stride=stride, dilation=dilation[0],
|
|
|
|
groups=groups, drop_block=drop_block)
|
|
|
|
groups=groups, drop_block=drop_block)
|
|
|
|