def forward(self, x):
    (_, _, down16, down32) = self.backbone(x)
    arm_down16 = self.arm16(down16)
    arm_down32 = self.arm32(down32)
    global_down32 = self.global_context(down32)
    global_down32 = F.interpolate(global_down32, size=down32.size()[2:], mode='bilinear', align_corners=True)
    arm_down32 = (arm_down32 + global_down32)
    arm_down32 = self.up32(arm_down32)
    arm_down32 = self.refine32(arm_down32)
    arm_down16 = (arm_down16 + arm_down32)
    arm_down16 = self.up16(arm_down16)
    arm_down16 = self.refine16(arm_down16)
    return (arm_down16, arm_down32)
def forward(self, x):
    (_, _, down16, down32) = self.backbone(x)
    arm_down16 = self.arm16(down16)
    arm_down32 = self.arm32(down32)
    global_down32 = self.global_context(down32)
    global_down32 = F.interpolate(global_down32, size=down32.size()[2:], mode='bilinear', align_corners=True)
    arm_down32 = (arm_down32 + global_down32)
    arm_down32 = self.up32(arm_down32)
    arm_down32 = self.refine32(arm_down32)
    arm_down16 = (arm_down16 + arm_down32)
    arm_down16 = self.up16(arm_down16)
    arm_down16 = self.refine16(arm_down16)
    return (arm_down16, arm_down32)