SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix md file to avoid evaluation crash (#10962)

Browse Source
This commit is contained in:
Yih-Dar
2021-03-30 20:26:22 +02:00
committed by GitHub
parent 3e09d813aa
commit e031162a6b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/research_projects/wav2vec2/FINE_TUNE_XLSR_WAV2VEC2.md
View File

@@ -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():
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"])
``` ```