sanchit-gandhi HF staff commited on
Commit
3075678
1 Parent(s): 3354f91

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: 4.2585
19
+ - Wer: 1.9292
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
+ | 4.7833 | 0.42 | 500 | 4.5563 | 1.8867 |
55
+ | 4.311 | 0.84 | 1000 | 4.2585 | 1.9292 |
56
 
57
 
58
  ### Framework versions