def __init__(self, reduction):
super().__init__()
self.conv1 = nn.Conv2d(3, 3, kernel_size=1)
self.conv2 = nn.Conv2d(3, 3, kernel_size=1)
self.input = nn.InputChoice(2, n_chosen=2, reduction=reduction)
def __init__(self):
super().__init__()
self.conv1 = nn.Conv2d(3, 3, kernel_size=1)
self.conv2 = nn.Conv2d(3, 5, kernel_size=1)
self.input = nn.InputChoice(2)