f908671185394de7527a18bb5889dc21a167c4bd,memcnn/models/additive.py,AdditiveBlockFunction,backward,#Any#Any#,148

Before Change

        // restore input
        xout = torch.cat([x1, x2], dim=1).contiguous()
        x.storage().resize_(int(np.prod(xout.shape)))
        x.set_(xout)

        return (grad_input, None, None) + FWgrads + GWgrads

After Change

        // restore input
        xout = torch.cat([x1, x2], dim=1).contiguous()
        with torch.no_grad():
            x.storage().resize_(int(np.prod(xout.shape)))
            x.set_(xout)

        return (grad_input, None, None) + FWgrads + GWgrads


class AdditiveBlockInverseFunction(torch.autograd.Function):

Instances