parent
e6d8f2ff4d
commit
a3aa741c87
|
@ -41,4 +41,8 @@ with torch.no_grad():
|
||||||
pred_ids = torch.argmax(logits, dim=-1).cpu()
|
pred_ids = torch.argmax(logits, dim=-1).cpu()
|
||||||
pred_text = processor.batch_decode(pred_ids)
|
pred_text = processor.batch_decode(pred_ids)
|
||||||
print('Transcription:', pred_text)
|
print('Transcription:', pred_text)
|
||||||
```
|
```
|
||||||
|
|
||||||
|
# Finetune Code
|
||||||
|
Github Repo:
|
||||||
|
https://github.com/CassiniHuy/wav2vec2_finetune
|
Loading…
Reference in New Issue