From d885f574ba077e693ad4c735b3cf9aa1b5bfb7af Mon Sep 17 00:00:00 2001 From: Mingrui Zhang Date: Sun, 4 Nov 2018 19:58:50 -0800 Subject: [PATCH] Fix evaluate input length Fix the length (originally it's char length) --- seq2seq-translation/seq2seq-translation-batched.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/seq2seq-translation/seq2seq-translation-batched.ipynb b/seq2seq-translation/seq2seq-translation-batched.ipynb index 465ce6f..46ddb9c 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)]\n", " input_batches = Variable(torch.LongTensor(input_seqs), volatile=True).transpose(0, 1)\n", " \n", " if USE_CUDA:\n",