Steven Liu commited on
Commit
7e5a2e8
1 Parent(s): 7edbb72

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -8
README.md CHANGED
@@ -16,8 +16,8 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 2.8796
20
- - Wer: 0.9833
21
 
22
  ## Model description
23
 
@@ -36,7 +36,7 @@ More information needed
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
- - learning_rate: 1e-05
40
  - train_batch_size: 8
41
  - eval_batch_size: 8
42
  - seed: 42
@@ -45,17 +45,15 @@ The following hyperparameters were used during training:
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
  - lr_scheduler_warmup_steps: 500
48
- - training_steps: 4000
49
  - mixed_precision_training: Native AMP
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Wer |
54
  |:-------------:|:-----:|:----:|:---------------:|:------:|
55
- | 2.8346 | 200.0 | 1000 | 2.9858 | 1.0 |
56
- | 2.4286 | 400.0 | 2000 | 2.8819 | 0.9958 |
57
- | 2.133 | 600.0 | 3000 | 2.9096 | 0.9792 |
58
- | 2.0029 | 800.0 | 4000 | 2.8796 | 0.9833 |
59
 
60
 
61
  ### Framework versions
 
16
 
17
  This model is a fine-tuned version of [facebook/wav2vec2-base](https://huggingface.co/facebook/wav2vec2-base) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 6.8626
20
+ - Wer: 1.0299
21
 
22
  ## Model description
23
 
 
36
  ### Training hyperparameters
37
 
38
  The following hyperparameters were used during training:
39
+ - learning_rate: 0.0001
40
  - train_batch_size: 8
41
  - eval_batch_size: 8
42
  - seed: 42
 
45
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
  - lr_scheduler_type: linear
47
  - lr_scheduler_warmup_steps: 500
48
+ - training_steps: 2000
49
  - mixed_precision_training: Native AMP
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Wer |
54
  |:-------------:|:-----:|:----:|:---------------:|:------:|
55
+ | 0.7266 | 499.8 | 1000 | 5.8888 | 0.9403 |
56
+ | 0.166 | 999.8 | 2000 | 6.8626 | 1.0299 |
 
 
57
 
58
 
59
  ### Framework versions