From aafb0b7514dbfc578419df01e798d071ab1accd3 Mon Sep 17 00:00:00 2001 From: Tad-Done Date: Mon, 2 Nov 2020 14:50:45 +0800 Subject: [PATCH 1/2] Update seq2seq-translation-batched.ipynb Fix a bug in evaluate() --- 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", From 36e81bb6fa832519b967bb2584186bf8e2b542c7 Mon Sep 17 00:00:00 2001 From: Tad-Done Date: Mon, 2 Nov 2020 15:47:42 +0800 Subject: [PATCH 2/2] Update seq2seq-translation-batched.ipynb --- 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 46ddb9c..19531ea 100644 --- a/seq2seq-translation/seq2seq-translation-batched.ipynb +++ b/seq2seq-translation/seq2seq-translation-batched.ipynb @@ -1290,7 +1290,7 @@ "source": [ "def evaluate(input_seq, max_length=MAX_LENGTH):\n", " input_seqs = [indexes_from_sentence(input_lang, input_seq)]\n", - " input_lengths = [len(input_seqs)]\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",