111

class Fusion_Group(nn.Module):
    def __init__(self, conv, dim, kernel_size):
        super(Fusion_Group, self).__init__()
        modules = [Block(conv, dim * 2, kernel_size)]
        self.gp = nn.Sequential(*modules)

    def forward(self, x):
        res = self.gp(x)
        res += x
        return res