diff --git a/seq2seq-translation/seq2seq-translation-batched.ipynb b/seq2seq-translation/seq2seq-translation-batched.ipynb index 465ce6f..19531ea 100644 --- a/seq2seq-translation/seq2seq-translation-batched.ipynb +++ b/seq2seq-translation/seq2seq-translation-batched.ipynb @@ -1289,8 +1289,8 @@ "outputs": [], "source": [ "def evaluate(input_seq, max_length=MAX_LENGTH):\n", - " input_lengths = [len(input_seq)]\n", " input_seqs = [indexes_from_sentence(input_lang, input_seq)]\n", + " input_lengths = [len(input_seqs[0])]\n", " input_batches = Variable(torch.LongTensor(input_seqs), volatile=True).transpose(0, 1)\n", " \n", " if USE_CUDA:\n",