def __init__(self, conv=common.default_conv):
    super(MSRB, self).__init__()
    n_feats = 64
    kernel_size_1 = 3
    kernel_size_2 = 5
    self.conv_3_1 = conv(n_feats, n_feats, kernel_size_1)
    self.conv_3_2 = conv((n_feats * 2), (n_feats * 2), kernel_size_1)
    self.conv_5_1 = conv(n_feats, n_feats, kernel_size_2)
    self.conv_5_2 = conv((n_feats * 2), (n_feats * 2), kernel_size_2)
    self.confusion = nn.Conv2d((n_feats * 4), n_feats, 1, padding=0, stride=1)
    self.relu = nn.ReLU(inplace=True)
def __init__(self, conv=common.default_conv):
    super(MSRB, self).__init__()
    n_feats = 64
    kernel_size_1 = 3
    kernel_size_2 = 5
    self.conv_3_1 = conv(n_feats, n_feats, kernel_size_1)
    self.conv_3_2 = conv((n_feats * 2), (n_feats * 2), kernel_size_1)
    self.conv_5_1 = conv(n_feats, n_feats, kernel_size_2)
    self.conv_5_2 = conv((n_feats * 2), (n_feats * 2), kernel_size_2)
    self.confusion = nn.Conv2d((n_feats * 4), n_feats, 1, padding=0, stride=1)
    self.relu = nn.ReLU(inplace=True)