Isa0 commited on
Commit
f67621d
·
1 Parent(s): df4889e

fix arguments in wer() functions

Browse files
Files changed (1) hide show
  1. train.py +2 -1
train.py CHANGED
@@ -66,7 +66,8 @@ def compute_metrics(pred):
66
  label_ids[label_ids == -100] = tokenizer.pad_token_id
67
  label_str = tokenizer.batch_decode(label_ids, group_tokens=False)
68
 
69
- error = wer(predictions=pred_str, references=label_str)
 
70
  return {"wer": error}
71
 
72
 
 
66
  label_ids[label_ids == -100] = tokenizer.pad_token_id
67
  label_str = tokenizer.batch_decode(label_ids, group_tokens=False)
68
 
69
+ # jiwer.wer(reference, hypothesis) — positional only
70
+ error = wer(label_str, pred_str)
71
  return {"wer": error}
72
 
73