Merge pull request #89 from tejaslodaya/master

Fix NLL_loss error

Author: spro

seq2seq-translation/seq2seq-translation.ipynb

@@ -823,14 +823,14 @@
     "        # Teacher forcing: Use the ground-truth target as the next input\n",
     "        for di in range(target_length):\n",
     "            decoder_output, decoder_context, decoder_hidden, decoder_attention = decoder(decoder_input, decoder_context, decoder_hidden, encoder_outputs)\n",
-    "            loss += criterion(decoder_output[0], target_variable[di])\n",
+    "            loss += criterion(decoder_output, target_variable[di])\n",
     "            decoder_input = target_variable[di] # Next target is next input\n",
     "\n",
     "    else:\n",
     "        # Without teacher forcing: use network's own prediction as the next input\n",
     "        for di in range(target_length):\n",
     "            decoder_output, decoder_context, decoder_hidden, decoder_attention = decoder(decoder_input, decoder_context, decoder_hidden, encoder_outputs)\n",
-    "            loss += criterion(decoder_output[0], target_variable[di])\n",
+    "            loss += criterion(decoder_output, target_variable[di])\n",
     "            \n",
     "            # Get most likely word index (highest value) from output\n",
     "            topv, topi = decoder_output.data.topk(1)\n",