fix arguments in wer() functions
Browse files
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 |
-
|
|
|
|
| 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 |
|