Update generate_seq2seq.py (#384)

dbiir · Aug 24, 2023 · 805ab4c · 805ab4c
1 parent 461373d
commit 805ab4c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/generate_seq2seq.py b/scripts/generate_seq2seq.py
@@ -29,11 +29,11 @@ def __init__(self, args):
         self.encoder = str2encoder[args.encoder](args)
         self.tgt_embedding = Embedding(args)
         for embedding_name in args.tgt_embedding:
-            tmp_emb = str2embedding[embedding_name](args, len(args.tokenizer.vocab))
+            tmp_emb = str2embedding[embedding_name](args, len(args.tgt_tokenizer.vocab))
             self.tgt_embedding.update(tmp_emb, embedding_name)
         self.decoder = str2decoder[args.decoder](args)
         self.target = Target()
-        self.target.update(LmTarget(args, len(args.tokenizer.vocab)), "lm")
+        self.target.update(LmTarget(args, len(args.tgt_tokenizer.vocab)), "lm")
 
     def forward(self, src, seg, tgt):
         emb = self.embedding(src, seg)