210c33dd05e3b84aaaf6e1430777bc1833216d99,fasterai/generators.py,Unet34,_pad,#Unet34#Any#Any#,54

Before Change

        target_w = target.shape[3]*2

        if h<target_h or w<target_w:
            target = Variable(torch.zeros(x.shape[0], x.shape[1], target_h, target_w))
            target[:,:,:h,:w]=x
            return to_gpu(target)

        return x

After Change

        if h<target_h or w<target_w:
            padh = target_h-h if target_h > h else 0
            padw = target_w-w if target_w > w else 0
            return F.pad(x, (0,padw,0,padh), "constant",0)

        return x
           
    def forward(self, x_in: torch.Tensor):

Instances