38c6a56381f2f93fd0fb819f81be26df86a6a0e1,python/baseline/pytorch/seq2seq/model.py,Seq2SeqAttnModel,__init__,#Seq2SeqAttnModel#Any#Any#,246

Before Change

        self.preds = nn.Linear(self.hsz, self.nc)
        self.probs = nn.LogSoftmax(dim=1)
        self.output_to_attn = nn.Linear(self.hsz, self.hsz, bias=False)
        self.attn_softmax = nn.Softmax(dim=1)
        self.attn_out = nn.Linear(2 * self.hsz, self.hsz, bias=False)
        self.attn_tanh = pytorch_activation("tanh")
        self.nlayers = nlayers

After Change

        self.probs = nn.LogSoftmax(dim=1)
        self.nlayers = nlayers
        attn_type = kwargs.get("attn_type", "luong").lower()
        if attn_type == "dot":
            self.attn_module = LuongDotProductAttention(enc_hsz)
        else:
            self.attn_module = LuongGeneralAttention(enc_hsz)

    def attn(self, output_t, context, src_mask=None):
        return self.attn_module(output_t, context, context, src_mask)

    def bridge(self, final_encoder_state, context):

Instances