Create README.md

README.md

@@ -0,0 +1,30 @@
+## Test model
+
+To test this model run the following code:
+
+```python
+from datasets import load_dataset
+from transformers import Wav2Vec2ForCTC
+import torchaudio
+import torch
+
+ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
+
+model = Wav2Vec2ForCTC.from_pretrained("patrickvonplaten/wav2vec2_tiny_random_robust")
+
+def load_audio(batch):
+    batch["samples"], _ = torchaudio.load(batch["file"])
+    return batch
+    
+ds = ds.map(load_audio)
+
+input_ids = torch.nn.utils.rnn.pad_sequence([torch.tensor(x[0]) for x in ds["samples"][:10]], batch_first=True)
+
+# forward
+logits = model(input_ids).logits
+pred_ids = torch.argmax(logits, dim=-1)
+
+# dummy loss
+dummy_labels = torch.zeros((logits.shape[0], logits.shape[0] - 10))
+loss = model(input_ids, labels=dummy_labels).loss
+```