Skip to content

GitLab

Unverified Commit e031162a authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

fix md file to avoid evaluation crash (#10962)

parent 3e09d813
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/research_projects/wav2vec2/FINE_TUNE_XLSR_WAV2VEC2.md
View file @ e031162a
...@@ -349,7 +349,7 @@ def speech_file_to_array_fn(batch): ...@@ -349,7 +349,7 @@ def speech_file_to_array_fn(batch):
return batch return batch
test_dataset = test_dataset.map(speech_file_to_array_fn) test_dataset = test_dataset.map(speech_file_to_array_fn)
inputs = processor(test_dataset["speech"][:2], sampling_rate=16_000, return_tensors="pt", padding=True) inputs = processor(test_dataset[:2]["speech"], sampling_rate=16_000, return_tensors="pt", padding=True)
with torch.no_grad(): with torch.no_grad():
logits = model(inputs.input_values, attention_mask=inputs.attention_mask).logits logits = model(inputs.input_values, attention_mask=inputs.attention_mask).logits
...@@ -357,7 +357,7 @@ with torch.no_grad(): ...@@ -357,7 +357,7 @@ with torch.no_grad():
predicted_ids = torch.argmax(logits, dim=-1) predicted_ids = torch.argmax(logits, dim=-1)
print("Prediction:", processor.batch_decode(predicted_ids)) print("Prediction:", processor.batch_decode(predicted_ids))
print("Reference:", test_dataset["sentence"][:2]) print("Reference:", test_dataset[:2]["sentence"])
``` ```
......
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