def __init__(self, layer, in_size, ch, out_size, stride=2):
    super(Block, self).__init__()
    self.add_link(BottleNeckA(in_size, ch, out_size, stride))
    for i in range((layer - 1)):
        self.add_link(BottleNeckB(out_size, ch))
    self._layer = layer
def __init__(self, layer, in_size, ch, out_size, stride=2):
    super(Block, self).__init__()
    self.add_link(BottleNeckA(in_size, ch, out_size, stride))
    for i in range((layer - 1)):
        self.add_link(BottleNeckB(out_size, ch))
    self._layer = layer