def forward(self, x):
    x = x.float()
    x = self.firstconv(x)
    x = self.firstbn(x)
    x = self.firstrelu(x)
    e1 = self.encoder1(x)
    e2 = self.encoder2(e1)
    e3 = self.encoder3(e2)
    e4 = self.encoder4(e3)
    e5 = self.center(e4)
    d5 = torch.cat([self.decoder5(e5), e4], 1)
    d5 = self.conv5(d5)
    d4 = torch.cat([self.decoder4(d5), e3], 1)
    d4 = self.conv4(d4)
    d3 = torch.cat([self.decoder3(d4), e2], 1)
    d3 = self.conv3(d3)
    d2 = torch.cat([self.decoder2(d3), e1], 1)
    d2 = self.conv2(d2)
    f = self.finaldeconv1(d2)
    f = self.logit(f)
    return f
def forward(self, x):
    x = x.float()
    x = self.firstconv(x)
    x = self.firstbn(x)
    x = self.firstrelu(x)
    e1 = self.encoder1(x)
    e2 = self.encoder2(e1)
    e3 = self.encoder3(e2)
    e4 = self.encoder4(e3)
    e5 = self.center(e4)
    d5 = torch.cat([self.decoder5(e5), e4], 1)
    d5 = self.conv5(d5)
    d4 = torch.cat([self.decoder4(d5), e3], 1)
    d4 = self.conv4(d4)
    d3 = torch.cat([self.decoder3(d4), e2], 1)
    d3 = self.conv3(d3)
    d2 = torch.cat([self.decoder2(d3), e1], 1)
    d2 = self.conv2(d2)
    f = self.finaldeconv1(d2)
    f = self.logit(f)
    return f