From 6024accadd65727f793ec39c8f7fc639efd54669 Mon Sep 17 00:00:00 2001
From: Leif Eriksson <leif.eriksson@liu.se>
Date: Fri, 10 Feb 2023 17:54:08 +0000
Subject: [PATCH] Corrected vocab usage in training-loop to avoid potential
collisions with earlier tests.
---
labs/l3/NLP-L3.ipynb | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/labs/l3/NLP-L3.ipynb b/labs/l3/NLP-L3.ipynb
index 960a939..ade668f 100644
--- a/labs/l3/NLP-L3.ipynb
+++ b/labs/l3/NLP-L3.ipynb
@@ -907,7 +907,7 @@
" train_loader = DataLoader(train_dataset, batch_size, shuffle=True, collate_fn=batcher)\n",
"\n",
" # Build the translator\n",
- " translator = Translator(src_vocab, tgt_vocab, ScaledDotProductAttention(), device=device)\n",
+ " translator = Translator(vocab_src, vocab_tgt, ScaledDotProductAttention(), device=device)\n",
"\n",
" # Initialise the optimiser\n",
" optimizer = torch.optim.Adam(translator.model.parameters(), lr=lr)\n",
@@ -929,7 +929,7 @@
"\n",
" # Forward pass\n",
" scores = translator.model(src_batch, tgt_batch_shifted)\n",
- " scores = scores.view(-1, len(tgt_vocab))\n",
+ " scores = scores.view(-1, len(vocab_tgt))\n",
"\n",
" # Backward pass\n",
" optimizer.zero_grad()\n",
--
GitLab