From 0f226f78ce49be39c355190f33482b5fcbedd01c Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 22 Mar 2021 10:32:21 +0300 Subject: [PATCH] push (#10846) --- examples/research_projects/wav2vec2/FINE_TUNE_XLSR_WAV2VEC2.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/research_projects/wav2vec2/FINE_TUNE_XLSR_WAV2VEC2.md b/examples/research_projects/wav2vec2/FINE_TUNE_XLSR_WAV2VEC2.md index 8fe27f257f..18847b8718 100644 --- a/examples/research_projects/wav2vec2/FINE_TUNE_XLSR_WAV2VEC2.md +++ b/examples/research_projects/wav2vec2/FINE_TUNE_XLSR_WAV2VEC2.md @@ -401,7 +401,7 @@ def evaluate(batch): with torch.no_grad(): 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) return batch