def forward(self, x, e=None):
    x = self.up(x)
    if (e is not None):
        x = torch.cat([x, e], 1)
    x = self.conv1(x)
    x = self.norm1(x)
    x = self.relu1(x)
    x = self.conv2(x)
    x = self.norm2(x)
    x = self.relu2(x)
    x = self.SCSE(x)
    return x
def forward(self, x, e=None):
    x = self.up(x)
    if (e is not None):
        x = torch.cat([x, e], 1)
    x = self.conv1(x)
    x = self.norm1(x)
    x = self.relu1(x)
    x = self.conv2(x)
    x = self.norm2(x)
    x = self.relu2(x)
    x = self.SCSE(x)
    return x