def __init__(self, layer, in_size, ch, out_size, stride=2, act=F.elu):
super(Block, self).__init__()
links = [('a', BottleNeckA(in_size, ch, out_size, stride, act))]
for i in range(layer-1):
links += [('b{}'.format(i+1), BottleNeckB(out_size, ch, act))]
for link in links:
self.add_link(*link)
self.forward = links
评论列表
文章目录