Skip to content

GitLab

Unverified Commit 395ffcd7 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix run seq2seq (#10547)

parent 54e55b52
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/research_projects/wav2vec2/run_asr.py
View file @ 395ffcd7
...@@ -251,7 +251,7 @@ def main(): ...@@ -251,7 +251,7 @@ def main():
pred_logits = pred.predictions pred_logits = pred.predictions
pred_ids = np.argmax(pred_logits, axis=-1) 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) pred_str = processor.batch_decode(pred_ids)
# we do not want to group tokens when computing the metrics # we do not want to group tokens when computing the metrics
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment