sanchit-gandhi HF staff commited on
Commit
205593a
1 Parent(s): bfc3753

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -4
README.md CHANGED
@@ -14,6 +14,9 @@ should probably proofread and complete it, then remove this comment. -->
14
  #
15
 
16
  This model was trained from scratch on the librispeech_asr dataset.
 
 
 
17
 
18
  ## Model description
19
 
@@ -33,11 +36,11 @@ More information needed
33
 
34
  The following hyperparameters were used during training:
35
  - learning_rate: 0.0001
36
- - train_batch_size: 8
37
- - eval_batch_size: 8
38
  - seed: 42
39
- - gradient_accumulation_steps: 8
40
- - total_train_batch_size: 64
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
43
  - lr_scheduler_warmup_steps: 500
@@ -46,6 +49,10 @@ The following hyperparameters were used during training:
46
 
47
  ### Training results
48
 
 
 
 
 
49
 
50
 
51
  ### Framework versions
 
14
  #
15
 
16
  This model was trained from scratch on the librispeech_asr dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 6.6618
19
+ - Wer: 1.8028
20
 
21
  ## Model description
22
 
 
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0001
39
+ - train_batch_size: 12
40
+ - eval_batch_size: 12
41
  - seed: 42
42
+ - gradient_accumulation_steps: 2
43
+ - total_train_batch_size: 24
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
  - lr_scheduler_warmup_steps: 500
 
49
 
50
  ### Training results
51
 
52
+ | Training Loss | Epoch | Step | Validation Loss | Wer |
53
+ |:-------------:|:-----:|:----:|:---------------:|:------:|
54
+ | 6.5791 | 0.42 | 500 | 6.8169 | 1.5493 |
55
+ | 6.3192 | 0.84 | 1000 | 6.6618 | 1.8028 |
56
 
57
 
58
  ### Framework versions