sanchit-gandhi commited on
Commit
a9d7e93
·
1 Parent(s): 59e022a

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -7
README.md CHANGED
@@ -15,8 +15,8 @@ should probably proofread and complete it, then remove this comment. -->
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: 5.5019
19
- - Wer: 1.5265
20
 
21
  ## Model description
22
 
@@ -39,8 +39,8 @@ The following hyperparameters were used during training:
39
  - train_batch_size: 8
40
  - eval_batch_size: 8
41
  - seed: 42
42
- - gradient_accumulation_steps: 2
43
- - total_train_batch_size: 16
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
  - lr_scheduler_warmup_steps: 500
@@ -51,9 +51,7 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Wer |
53
  |:-------------:|:-----:|:----:|:---------------:|:------:|
54
- | 6.7441 | 0.28 | 500 | 6.9081 | 0.9404 |
55
- | 6.7273 | 0.56 | 1000 | 6.7867 | 1.9963 |
56
- | 5.6598 | 0.84 | 1500 | 5.5019 | 1.5265 |
57
 
58
 
59
  ### Framework versions
 
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.8224
19
+ - Wer: 0.9494
20
 
21
  ## Model description
22
 
 
39
  - train_batch_size: 8
40
  - eval_batch_size: 8
41
  - seed: 42
42
+ - gradient_accumulation_steps: 4
43
+ - total_train_batch_size: 32
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
  - lr_scheduler_warmup_steps: 500
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Wer |
53
  |:-------------:|:-----:|:----:|:---------------:|:------:|
54
+ | 6.4267 | 0.56 | 500 | 6.8224 | 0.9494 |
 
 
55
 
56
 
57
  ### Framework versions