TKU410410103 commited on
Commit
1e8e38d
1 Parent(s): de7c257

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -50,13 +50,14 @@ The training hyperparameters remained consistent throughout the fine-tuning proc
50
  - eval_batch_size: 16
51
  - seed: 42
52
  - gradient_accumulation_steps: 2
53
- - num_train_epochs: 30
 
54
  - lr_scheduler_type: linear
55
 
56
  ### How to evaluate the model
57
 
58
  ```python
59
- from transformers import HubertForCTC, Wav2Vec2Processor
60
  from datasets import load_dataset
61
  import torchaudio
62
  import librosa
 
50
  - eval_batch_size: 16
51
  - seed: 42
52
  - gradient_accumulation_steps: 2
53
+ - num_train_epochs: 35
54
+ - warmup_steps: 2000
55
  - lr_scheduler_type: linear
56
 
57
  ### How to evaluate the model
58
 
59
  ```python
60
+ from transformers import Wav2vec2ForCTC, Wav2Vec2Processor
61
  from datasets import load_dataset
62
  import torchaudio
63
  import librosa