def __init__(self, layer, in_size, ch, out_size, stride=1):
super(LaminationBlock, self).__init__()
links = [('lb0', Block(in_size, ch, out_size, stride))]
links += [('lb{}'.format(i+1),
Block(out_size, ch, out_size, 1)) for i in range(1, layer)]
for l in links:
self.add_link(*l)
self.forward = links
评论列表
文章目录