azuur commited on
Commit
714af6c
1 Parent(s): 9a742bc

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +18 -2
README.md CHANGED
@@ -14,7 +14,10 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # wav2vec2-base-gn-demo
16
 
17
- This model is a fine-tuned version of [facebook/wav2vec2-xls-r-300m](https://huggingface.co/facebook/wav2vec2-xls-r-300m) on the common_voice dataset.
 
 
 
18
 
19
  ## Model description
20
 
@@ -34,7 +37,7 @@ More information needed
34
 
35
  The following hyperparameters were used during training:
36
  - learning_rate: 0.0002
37
- - train_batch_size: 32
38
  - eval_batch_size: 8
39
  - seed: 42
40
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
@@ -43,6 +46,19 @@ The following hyperparameters were used during training:
43
  - num_epochs: 30
44
  - mixed_precision_training: Native AMP
45
 
 
 
 
 
 
 
 
 
 
 
 
 
 
46
  ### Framework versions
47
 
48
  - Transformers 4.11.3
 
14
 
15
  # wav2vec2-base-gn-demo
16
 
17
+ This model is a fine-tuned version of [azuur/wav2vec2-base-gn-demo](https://huggingface.co/azuur/wav2vec2-base-gn-demo) on the common_voice dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 0.6992
20
+ - Wer: 0.7439
21
 
22
  ## Model description
23
 
 
37
 
38
  The following hyperparameters were used during training:
39
  - learning_rate: 0.0002
40
+ - train_batch_size: 16
41
  - eval_batch_size: 8
42
  - seed: 42
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 
46
  - num_epochs: 30
47
  - mixed_precision_training: Native AMP
48
 
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss | Wer |
52
+ |:-------------:|:-----:|:----:|:---------------:|:------:|
53
+ | No log | 4.0 | 100 | 2.4470 | 1.0015 |
54
+ | No log | 8.0 | 200 | 0.9011 | 0.8720 |
55
+ | No log | 12.0 | 300 | 0.8527 | 0.9177 |
56
+ | No log | 16.0 | 400 | 0.7743 | 0.7988 |
57
+ | 1.0068 | 20.0 | 500 | 0.7984 | 0.8049 |
58
+ | 1.0068 | 24.0 | 600 | 0.7133 | 0.7530 |
59
+ | 1.0068 | 28.0 | 700 | 0.6992 | 0.7439 |
60
+
61
+
62
  ### Framework versions
63
 
64
  - Transformers 4.11.3