shields commited on
Commit
f7e9d4d
1 Parent(s): 3c86eff

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -8
README.md CHANGED
@@ -14,13 +14,8 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the None dataset.
16
  It achieves the following results on the evaluation set:
17
- - eval_loss: 0.5215
18
- - eval_wer: 0.2411
19
- - eval_runtime: 54.0563
20
- - eval_samples_per_second: 5.272
21
- - eval_steps_per_second: 0.666
22
- - epoch: 25.25
23
- - step: 5000
24
 
25
  ## Model description
26
 
@@ -46,9 +41,22 @@ The following hyperparameters were used during training:
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_steps: 1000
49
- - num_epochs: 50
50
  - mixed_precision_training: Native AMP
51
 
 
 
 
 
 
 
 
 
 
 
 
 
 
52
  ### Framework versions
53
 
54
  - Transformers 4.11.3
 
14
 
15
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the None dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Loss: 0.4338
18
+ - Wer: 0.2313
 
 
 
 
 
19
 
20
  ## Model description
21
 
 
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
43
  - lr_scheduler_warmup_steps: 1000
44
+ - num_epochs: 20
45
  - mixed_precision_training: Native AMP
46
 
47
+ ### Training results
48
+
49
+ | Training Loss | Epoch | Step | Validation Loss | Wer |
50
+ |:-------------:|:-----:|:----:|:---------------:|:------:|
51
+ | 3.6839 | 2.53 | 500 | 2.7287 | 1.0 |
52
+ | 0.8708 | 5.05 | 1000 | 0.5004 | 0.3490 |
53
+ | 0.2879 | 7.58 | 1500 | 0.4411 | 0.2872 |
54
+ | 0.1877 | 10.1 | 2000 | 0.4359 | 0.2594 |
55
+ | 0.1617 | 12.63 | 2500 | 0.4404 | 0.2492 |
56
+ | 0.1295 | 15.15 | 3000 | 0.4356 | 0.2418 |
57
+ | 0.1146 | 17.68 | 3500 | 0.4338 | 0.2313 |
58
+
59
+
60
  ### Framework versions
61
 
62
  - Transformers 4.11.3