bug in beam search

ryankiros · Nov 3, 2015 · cd5d8bf · cd5d8bf
1 parent 6efbad6
commit cd5d8bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/search.py b/search.py
@@ -63,7 +63,7 @@ def gen_sample(tparams, f_init, f_next, ctx, options, trng=None, k=1, maxlen=30,
 
             for idx, [ti, wi] in enumerate(zip(trans_indices, word_indices)):
                 new_hyp_samples.append(hyp_samples[ti]+[wi])
-                new_hyp_scores[idx] = copy.copy(costs[ti])
+                new_hyp_scores[idx] = copy.copy(costs[idx])
                 new_hyp_states.append(copy.copy(next_state[ti]))
 
             # check the finished samples