def __init__(self, in_size, ch, out_size, stride, card=32):
super(Block, self).__init__()
links = [(
'c{}'.format(i+1),
Cardinality(in_size, ch, out_size, stride)) for i in range(card)]
links += [(
'x_bypass',
L.Convolution2D(in_size, out_size, 1, stride, 0, nobias=True))]
for l in links:
self.add_link(*l)
self.forward = links
评论列表
文章目录