def __init__(self, in_size, ch, act=F.elu):
w = math.sqrt(2)
super(BottleNeckB, self).__init__(
conv1=L.Convolution2D(in_size, ch, 1, 1, 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, in_size, 1, 1, 0, w, nobias=True),
bn3=L.BatchNormalization(in_size),
)
self.act=act
评论列表
文章目录