Skip to content

Commit 0f226f7

Browse files
push (#10846)
1 parent 82b8d8c commit 0f226f7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/research_projects/wav2vec2/FINE_TUNE_XLSR_WAV2VEC2.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -401,7 +401,7 @@ def evaluate(batch):
401401
with torch.no_grad():
402402
logits = model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits
403403

404-
pred_ids = torch.argmax(logits, dim=-1)
404+
pred_ids = torch.argmax(logits, dim=-1)
405405
batch["pred_strings"] = processor.batch_decode(pred_ids)
406406
return batch
407407

0 commit comments

Comments
 (0)