Fixed share_decoder_embeddings AttributeError (but maybe not the righ…

…t way)
surafelml · Jul 16, 2017 · 9fc0eb2 · 9fc0eb2
1 parent 7e809ba
commit 9fc0eb2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -369,7 +369,7 @@ def main():
             nn.Linear(opt.rnn_size, dicts['tgt'].size()),
             nn.LogSoftmax())
         if opt.share_decoder_embeddings:
-            generator[0].weight = decoder.word_lut.weight
+            generator[0].weight = decoder.embeddings.word_lut.weight
 
     model = onmt.Models.NMTModel(encoder, decoder, len(opt.gpus) > 1)