joaoalvarenga
/

wav2vec2-large-xlsr-portuguese

Automatic Speech Recognition

portuguese-speech-corpus

xlsr-fine-tuning-week

Inference Endpoints

Model card Files Files and versions Community

joaoalvarenga commited on Mar 25, 2021

Commit

eb9a32d

•

1 Parent(s): 0ab249c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -115,7 +115,7 @@ def evaluate(batch):
         logits = model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits
     pred_ids = torch.argmax(logits, dim=-1)
-    tbatch["pred_strings"] = processor.batch_decode(pred_ids)
     return batch
 result = test_dataset.map(evaluate, batched=True, batch_size=8)

         logits = model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits
     pred_ids = torch.argmax(logits, dim=-1)
+    batch["pred_strings"] = processor.batch_decode(pred_ids)
     return batch
 result = test_dataset.map(evaluate, batched=True, batch_size=8)