indonesian-nlp
/

wav2vec2-large-xlsr-indonesian

Automatic Speech Recognition Transformers PyTorch JAX

Indonesian wav2vec2 audio speech xlsr-fine-tuning-week Eval Results Inference Endpoints

Model card Files Files and versions Community

cahya commited on Mar 29, 2021

Commit

a542f83

•

1 Parent(s): 3758ee5

fixed the tabs/indentation

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -108,7 +108,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)
   batch["pred_strings"] = processor.batch_decode(pred_ids)
   return 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)
   batch["pred_strings"] = processor.batch_decode(pred_ids)
   return batch