hieuit7 commited on
Commit
273e4ce
1 Parent(s): cc5054f

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -14
README.md CHANGED
@@ -1,10 +1,6 @@
1
  ---
2
- language:
3
- - vi
4
  license: apache-2.0
5
  tags:
6
- - automatic-speech-recognition
7
- - common_voice
8
  - generated_from_trainer
9
  datasets:
10
  - common_voice
@@ -18,10 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  # wav2vec2-common_voice-vi
20
 
21
- This model is a fine-tuned version of [facebook/wav2vec2-large-xlsr-53](https://huggingface.co/facebook/wav2vec2-large-xlsr-53) on the COMMON_VOICE - VI dataset.
22
- It achieves the following results on the evaluation set:
23
- - Loss: 3.5260
24
- - Wer: 1.0
25
 
26
  ## Model description
27
 
@@ -41,21 +34,19 @@ More information needed
41
 
42
  The following hyperparameters were used during training:
43
  - learning_rate: 0.0003
44
- - train_batch_size: 16
45
  - eval_batch_size: 8
46
  - seed: 42
47
- - gradient_accumulation_steps: 2
48
- - total_train_batch_size: 32
49
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
  - lr_scheduler_type: linear
51
  - lr_scheduler_warmup_steps: 500
52
  - num_epochs: 15.0
 
53
 
54
  ### Training results
55
 
56
- | Training Loss | Epoch | Step | Validation Loss | Wer |
57
- |:-------------:|:-----:|:----:|:---------------:|:---:|
58
- | No log | 7.67 | 100 | 11.8240 | 1.0 |
59
 
60
 
61
  ### Framework versions
 
1
  ---
 
 
2
  license: apache-2.0
3
  tags:
 
 
4
  - generated_from_trainer
5
  datasets:
6
  - common_voice
 
14
 
15
  # wav2vec2-common_voice-vi
16
 
17
+ This model is a fine-tuned version of [facebook/wav2vec2-large-xlsr-53](https://huggingface.co/facebook/wav2vec2-large-xlsr-53) on the common_voice dataset.
 
 
 
18
 
19
  ## Model description
20
 
 
34
 
35
  The following hyperparameters were used during training:
36
  - learning_rate: 0.0003
37
+ - train_batch_size: 1
38
  - eval_batch_size: 8
39
  - seed: 42
40
+ - gradient_accumulation_steps: 200
41
+ - total_train_batch_size: 200
42
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
  - lr_scheduler_type: linear
44
  - lr_scheduler_warmup_steps: 500
45
  - num_epochs: 15.0
46
+ - mixed_precision_training: Native AMP
47
 
48
  ### Training results
49
 
 
 
 
50
 
51
 
52
  ### Framework versions