Fixed bug in rnnlm

clab · Dec 12, 2016 · 6859387 · 6859387
1 parent e2ed6f8
commit 6859387
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorial_rnnlm.py b/tutorial_rnnlm.py
@@ -78,7 +78,7 @@ def calc_lm_loss(sent):
     wids = [vw.w2i[w] for w in sent]
 
     # start the rnn by inputting "<s>"
-    s = f_init.add_input(wembs[-1]) 
+    s = f_init.add_input(WORDS_LOOKUP[wids[-1]]) 
 
     # feed word vectors into the RNN and predict the next word
     losses = []