patrickvonplaten commited on
Commit
bb9a3e9
1 Parent(s): 1970dc0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -23,7 +23,7 @@ results:
23
  metrics:
24
  - name: Test WER
25
  type: wer
26
- value: 0.45
27
  ---
28
 
29
  # Wav2Vec2-Large-XLSR-53-Mongolian
@@ -108,7 +108,7 @@ def evaluate(batch):
108
  with torch.no_grad():
109
  logits = model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits
110
 
111
- pred_ids = torch.argmax(logits, dim=-1)
112
  batch["pred_strings"] = processor.batch_decode(pred_ids)
113
  return batch
114
 
 
23
  metrics:
24
  - name: Test WER
25
  type: wer
26
+ value: 45.82
27
  ---
28
 
29
  # Wav2Vec2-Large-XLSR-53-Mongolian
 
108
  with torch.no_grad():
109
  logits = model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits
110
 
111
+ pred_ids = torch.argmax(logits, dim=-1)
112
  batch["pred_strings"] = processor.batch_decode(pred_ids)
113
  return batch
114