fix goldscore OpenNMT#1383 (OpenNMT#1423)

* fix OpenNMT#1383 * fix gold score only
AMusketeer · May 6, 2019 · c8c2369 · c8c2369
1 parent 03fe8d9
commit c8c2369
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onmt/translate/translator.py b/onmt/translate/translator.py
@@ -806,7 +806,7 @@ def _score_target(self, batch, memory_bank, src_lengths,
             memory_lengths=src_lengths, src_map=src_map)
 
         log_probs[:, :, self._tgt_pad_idx] = 0
-        gold = tgt_in
+        gold = tgt[1:]
         gold_scores = log_probs.gather(2, gold)
         gold_scores = gold_scores.sum(dim=0).view(-1)