lnxdx commited on
Commit
869d3c6
1 Parent(s): 8bed95e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -86,7 +86,7 @@ More information needed
86
 
87
  ## Training procedure
88
 
89
- ##### Model hyperparameters
90
  ```python
91
  model = Wav2Vec2ForCTC.from_pretrained(
92
  model_name_or_path if not last_checkpoint else last_checkpoint,
@@ -100,7 +100,7 @@ model = Wav2Vec2ForCTC.from_pretrained(
100
  ctc_zero_infinity = True,
101
  )
102
  ```
103
- ##### Training hyperparameters
104
 
105
  The following hyperparameters were used during training:
106
  - learning_rate: 1e-05
@@ -115,7 +115,7 @@ The following hyperparameters were used during training:
115
  - training_steps: 2000
116
  - mixed_precision_training: Native AMP
117
 
118
- ##### Training results
119
 
120
  | Training Loss | Epoch | Step | Validation Loss | Wer |
121
  |:-------------:|:-----:|:----:|:---------------:|:------:|
@@ -141,7 +141,7 @@ The following hyperparameters were used during training:
141
  | 0.7618 | 12.5 | 2000 | 0.6728 | 0.3286 |
142
 
143
 
144
- ##### Framework versions
145
 
146
  - Transformers 4.35.2
147
  - Pytorch 2.1.0+cu118
 
86
 
87
  ## Training procedure
88
 
89
+ #### Model hyperparameters
90
  ```python
91
  model = Wav2Vec2ForCTC.from_pretrained(
92
  model_name_or_path if not last_checkpoint else last_checkpoint,
 
100
  ctc_zero_infinity = True,
101
  )
102
  ```
103
+ #### Training hyperparameters
104
 
105
  The following hyperparameters were used during training:
106
  - learning_rate: 1e-05
 
115
  - training_steps: 2000
116
  - mixed_precision_training: Native AMP
117
 
118
+ #### Training results
119
 
120
  | Training Loss | Epoch | Step | Validation Loss | Wer |
121
  |:-------------:|:-----:|:----:|:---------------:|:------:|
 
141
  | 0.7618 | 12.5 | 2000 | 0.6728 | 0.3286 |
142
 
143
 
144
+ #### Framework versions
145
 
146
  - Transformers 4.35.2
147
  - Pytorch 2.1.0+cu118