def __init__(self, in_size, ch, out_size, stride=2, act=F.elu):
w = math.sqrt(2)
super(BottleNeckA, self).__init__(
conv1=L.Convolution2D(in_size, ch, 1, stride, 0, w, nobias=True),
bn1=L.BatchNormalization(ch),
conv2=L.Convolution2D(ch, ch, 3, 1, 1, w, nobias=True),
bn2=L.BatchNormalization(ch),
conv3=L.Convolution2D(ch, out_size, 1, 1, 0, w, nobias=True),
bn3=L.BatchNormalization(out_size),
conv4=L.Convolution2D(in_size, out_size, 1, stride, 0, w, nobias=True),
bn4=L.BatchNormalization(out_size),
)
self.act=act
评论列表
文章目录