def __init__(self): self.cb = ConvBn(2) self.cb2 = ConvBn(2) self.shared1 = Shared(self.cb2) self.shared2 = Shared(self.cb2)
def __init__(self): self.conv_bn_1 = ConvBn(1) self.conv_bn_2 = ConvBn(1)
def __init__(self): self.conv_bn = ConvBn(3)