fix attention in biaffine

dsraghav04 · Sep 30, 2019 · 83a13c3 · 83a13c3
1 parent 084db60
commit 83a13c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dependency-parser/7.biaffine-attention.ipynb b/dependency-parser/7.biaffine-attention.ipynb
@@ -407,7 +407,7 @@
     "    def forward(self, input_word, input_char, mask):\n",
     "        arcs, types, _ = self.encode(input_word, input_char)\n",
     "        \n",
-    "        out_arc = tf.squeeze(self.attention.forward(arcs[0], arcs[0], mask_d=mask, mask_e=mask), axis = 1)\n",
+    "        out_arc = tf.squeeze(self.attention.forward(arcs[0], arcs[1], mask_d=mask, mask_e=mask), axis = 1)\n",
     "        return out_arc, types, mask\n",
     "    \n",
     "    def loss(self, input_word, input_char, mask, heads, types):\n",