def __init__(self, in_channels, out_channels, inter_channels):
super().__init__()
self.in_channels = in_channels
self.out_channels = out_channels
self.inter_channels = inter_channels
self.conv1 = self._create_conv1()
self.dp1 = create_dropout()
self.conv2 = self._create_conv2()
self.dp2 = create_dropout()
def __init__(self):
super().__init__()
self.l1 = create_k1_conv(2, 2, bias=True, padding_mode='zeros')
self.l2 = create_norm(2)
self.l3 = create_activ()
self.l4 = create_dropout()
self.l5 = create_avg_pool(2)
self.l6 = create_two_upsample()
self.l7 = create_global_avg_pool()
def __init__(self):
super().__init__()
self.l1 = create_k3_conv(2, 2, stride=2, bias=False)
self.l2 = create_norm(2)
self.l3 = create_activ()
self.l4 = create_dropout()
self.l5 = create_two_avg_pool()
self.l6 = create_two_upsample()
self.l7 = create_global_avg_pool()
def __init__(self, in_channels, out_channels):
super().__init__()
self.in_channels = in_channels
self.out_channels = out_channels
self.conv = self._create_conv()
self.norm = self._create_norm()
self.activ = create_activ()
self.dp = create_dropout()
self.up = create_two_upsample()