def forward(self, x):
    residual = x
    main = self.relu(self.conv1_bn(self.compress_conv1(x)))
    main = self.relu(self.conv2_1_bn(self.conv2_1(main)))
    main = self.relu(self.conv2_2_bn(self.conv2_2(main)))
    main = self.conv3_bn(self.extend_conv3(main))
    main = self.regul(main)
    out = self.relu(torch.add(residual, main))
    return out
def forward(self, x):
    residual = x
    main = self.relu(self.conv1_bn(self.compress_conv1(x)))
    main = self.relu(self.conv2_1_bn(self.conv2_1(main)))
    main = self.relu(self.conv2_2_bn(self.conv2_2(main)))
    main = self.conv3_bn(self.extend_conv3(main))
    main = self.regul(main)
    out = self.relu(torch.add(residual, main))
    return out