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