be91aa08bda8873d839a77932049e39b0ee11577,models/common/biaffine.py,PairwiseBiaffineScorer,forward,#PairwiseBiaffineScorer#Any#Any#,63

Before Change

        self.W2.weight.data.zero_()

    def forward(self, input1, input2):
        res = self.W1(input1).unsqueeze(2) + self.W2(input2).unsqueeze(1) + self.W_bilin(input1, input2)
        return res

class DeepBiaffineScorer(nn.Module):
    def __init__(self, input1_size, input2_size, hidden_size, output_size, hidden_func=F.relu, dropout=0, pairwise=True):

After Change

        self.W_bilin.bias.data.zero_()

    def forward(self, input1, input2):
        input1 = torch.cat([input1, input1.new_ones(*input1.size()[:-1], 1)], len(input1.size())-1)
        input2 = torch.cat([input2, input2.new_ones(*input2.size()[:-1], 1)], len(input2.size())-1)
        return self.W_bilin(input1, input2)

class DeepBiaffineScorer(nn.Module):

Instances