Configurable alignment method

2026-09-09 11:16:00 +08:00 · 2018-05-23 06:17:01 -07:00
parent d8c460442a
commit 0f933106ca
1 changed files with 1 additions and 1 deletions
@@ -105,7 +105,7 @@ class AttentionRNN(nn.Module):
        # Alignment
        # (batch, max_time)
        # e_{ij} = a(s_{i-1}, h_j)
-        if attnetion_vec is None:
+        if self.align_model is 'b':
            alignment = self.alignment_model(annotations, rnn_output)
        else:
            alignment = self.alignment_model(annotations, rnn_output, attention_vec)