"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "0558c9cb9b62771d3e3955dc15d34475c2c86995"
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
......@@ -349,7 +349,7 @@ def speech_file_to_array_fn(batch):
return batch
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():
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)
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