Update model_attention.py

bug fix
yaoli · Nov 30, 2016 · 22e7647 · 22e7647
1 parent 97773d6
commit 22e7647
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model_attention.py b/model_attention.py
@@ -753,7 +753,7 @@ def gen_sample(self, tparams, f_init, f_next, ctx0, ctx_mask, options,
 
                 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])
                     for lidx in xrange(n_layers_lstm):
                         new_hyp_states[lidx].append(copy.copy(next_state[lidx][ti]))
                     for lidx in xrange(n_layers_lstm):