scasutt commited on
Commit
432ef58
·
1 Parent(s): ca3c586

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -7
README.md CHANGED
@@ -14,8 +14,8 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [scasutt/Prototype_training](https://huggingface.co/scasutt/Prototype_training) on the None dataset.
16
  It achieves the following results on the evaluation set:
17
- - Loss: 0.6686
18
- - Wer: 0.6639
19
 
20
  ## Model description
21
 
@@ -38,6 +38,8 @@ The following hyperparameters were used during training:
38
  - train_batch_size: 16
39
  - eval_batch_size: 8
40
  - seed: 42
 
 
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
43
  - lr_scheduler_warmup_steps: 1000
@@ -47,11 +49,8 @@ The following hyperparameters were used during training:
47
 
48
  | Training Loss | Epoch | Step | Validation Loss | Wer |
49
  |:-------------:|:-----:|:----:|:---------------:|:------:|
50
- | 0.6613 | 0.74 | 100 | 0.6388 | 0.6609 |
51
- | 0.5528 | 1.47 | 200 | 0.6396 | 0.6569 |
52
- | 0.5148 | 2.21 | 300 | 0.6362 | 0.6681 |
53
- | 0.4727 | 2.94 | 400 | 0.6537 | 0.6711 |
54
- | 0.4415 | 3.68 | 500 | 0.6686 | 0.6639 |
55
 
56
 
57
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [scasutt/Prototype_training](https://huggingface.co/scasutt/Prototype_training) on the None dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Loss: 0.5002
18
+ - Wer: 0.5111
19
 
20
  ## Model description
21
 
 
38
  - train_batch_size: 16
39
  - eval_batch_size: 8
40
  - seed: 42
41
+ - gradient_accumulation_steps: 2
42
+ - total_train_batch_size: 32
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
  - lr_scheduler_warmup_steps: 1000
 
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Wer |
51
  |:-------------:|:-----:|:----:|:---------------:|:------:|
52
+ | 6.7314 | 1.47 | 100 | 2.9310 | 0.9983 |
53
+ | 1.8053 | 2.94 | 200 | 0.5002 | 0.5111 |
 
 
 
54
 
55
 
56
  ### Framework versions