From 395ffcd757103ed2ccc888e48d90fd2ccb4d506f Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 5 Mar 2021 18:17:12 +0300 Subject: [PATCH] fix run seq2seq (#10547) --- examples/research_projects/wav2vec2/run_asr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/research_projects/wav2vec2/run_asr.py b/examples/research_projects/wav2vec2/run_asr.py index 00c64840e2..21144f58be 100755 --- a/examples/research_projects/wav2vec2/run_asr.py +++ b/examples/research_projects/wav2vec2/run_asr.py @@ -251,7 +251,7 @@ def main(): pred_logits = pred.predictions pred_ids = np.argmax(pred_logits, axis=-1) - pred.label_ids[pred.label_ids == -100] = 0 + pred.label_ids[pred.label_ids == -100] = processor.tokenizer.pad_token_id pred_str = processor.batch_decode(pred_ids) # we do not want to group tokens when computing the metrics