c3ec7981fd91befe035b667bed8ea538c8853b97,pytorch/pytorchcv/models/pspnet.py,PSPNet,forward,#PSPNet#Any#,168

Before Change

        x = self.backbone(x)
        x = self.pool(x)
        x = self.conv1(x)
        x = self.dropout(x)
        x = self.conv2(x)
        x = F.interpolate(x, size=(h, w), mode="bilinear", align_corners=True)
        return x


def get_pspnet(backbone,

After Change

        x = self.final_block(x)
        if self.aux:
            y = self.aux_block(y)
            return x, y
        else:
            return x

Instances