riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 10:41:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

push (#10846)

Browse Source
This commit is contained in:
Patrick von Platen 2021-03-22 10:32:21 +03:00 committed by GitHub
parent 82b8d8c7b0
commit 0f226f78ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -401,7 +401,7 @@ def evaluate(batch):
with torch.no_grad(): with torch.no_grad():
logits = model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits logits = model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits
pred_ids = torch.argmax(logits, dim=-1) pred_ids = torch.argmax(logits, dim=-1)
batch["pred_strings"] = processor.batch_decode(pred_ids) batch["pred_strings"] = processor.batch_decode(pred_ids)
return batch return batch